素因数分解, 約数の種類数, 高速版 osa_k ~高速な素因数分解~

// 素因数分解
map<ll, ll> factorize(ll n){
    map<ll, ll> mp;
    FOR(i, 2, n+1){
        while(n%i==0){
            mp[i]++;
            n/=i;
        }
    }
    return mp;
}
  • シンプルにできたので保存
  • 計算量O(N)

別VER

// 素因数分解
// その素因数が何個あるかのmapを返す
map<ll, ll> factorize(ll n){
    map<ll, ll> mp;
    ll sq = sqrt(n);
    FOR(i, 2, sq+1){
        while(n%i==0){
            mp[i]++;
            n/=i;
        }
    }
    // 残り
    if(n!=1){
        mp[n]++;
    }
    return mp;
}

// 約数の種類数
// 6 => 1, 2, 3, 6なので4
ll yaku(ll n){
  auto mp = factorize(n);
  ll ret = 1;
  for(auto pa : mp){
    ret *= (pa.second+1);
  }
  return ret;
}

SPF配列

素因数分解(高速版)O(log(N))

const int N_MAX = 2000000;
ll spf[N_MAX]; // smallest prime factors
void prepare_factorize(){
  rep(i,N_MAX) spf[i] = i;
  for(int p=2; p*p <= N_MAX; p++){
    for(int i=p; i<N_MAX; i+=p){
      if(spf[i]==i) spf[i] = p;
    }
  }
}

// 素因数分解
// その素因数が何個あるかのmapを返す
map<ll, ll> factorize_fast(ll n){
    if(spf[1]==0){
      p("please initialize");
      exit(0);
    }
    map<ll, ll> mp;
    while(n!=1){
      ll p = spf[n];
      mp[p]++;
      n/=p;
    }
    return mp;
}

// 約数の種類数
// 6 => 1, 2, 3, 6なので4
ll yaku(ll n){
  auto mp = factorize_fast(n);
  ll ret = 1;
  for(auto pa : mp){
    ret *= (pa.second+1);
  }
  return ret;
}

ll f(ll v){
  return v - yaku(v);
}

int main(){
    cin.tie(0);
    ios::sync_with_stdio(false);

    // input
    ll X;cin>>X;

    prepare_factorize();

    vector<pair<ll, PII>> V;
    FOR(a, 1, X){
      ll b = X-a;
      if(b<=0) continue;
      ll v = abs(f(a)-f(b));
      V.push_back(MP(v, MP(a,b)));
    }
    sort(ALL(V));

    ll mi = V[0].first;
    for(auto triplet : V){
      if(triplet.first==mi){
        ll a = triplet.second.first;
        ll b = triplet.second.second;
        p2(a,b);
      }
    }
    
    return 0;
}

高速素因数分解(logA)を約数列挙に適用する

f:id:peroon:20200615135034p:plain

// 高速約数列挙
// 名前は変えたほうがいいかもね
// devisorのためのdfsということでdfsd
void dfsd(ll cur_idx, ll cur_val, VI&Y, 
  vector<PII>& mp // (value, count)
  ){
  ll N = mp.size();
  if(cur_idx==N){
    // 値が完成
    Y.push_back(cur_val); return;
  }
  ll v = mp[cur_idx].first;
  ll c = mp[cur_idx].second;
  // p乗を全通り試す (0, ..., p乗)
  ll mul = 1;
  rep(p, c+1){
    dfsd(cur_idx+1, cur_val*mul, Y, mp);
    mul *= v;
  }
  return;
}

// 約数一覧 O(nlogn)
VI calc_devisors_fast(ll a){
  VI Y; // 戻り値(入れ物)
  
  auto mp = factorize_fast(a);
  vector<PII> V;
  for(auto pa : mp){
    V.push_back(pa);
  }

  dfsd(0, 1, Y, V);
  return Y;
}