#93709 (C++11) No.278 連続する整数の和(2)

• 問題: No.278 連続する整数の和(2)
• ユーザー: どらら
• 提出日時: 2016-05-23 23:42:29
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 12 ms / 2,000 ms
• コード長: 1,557 bytes
• コンパイル時間: 934 ms
• コンパイル使用メモリ: 96,452 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-02 11:53:48
• 合計ジャッジ時間: 1,612 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	12 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
testcase_06	AC	1 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	1 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	1 ms; 5,376 KB
testcase_11	AC	3 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	12 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	7 ms; 5,376 KB
testcase_16	AC	2 ms; 5,376 KB
testcase_17	AC	2 ms; 5,376 KB

ソースコード

#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cctype>
#include <cmath>
#include <iostream>
#include <queue>
#include <list>
#include <stack>
#include <map>
#include <numeric>
#include <set>
#include <sstream>
#include <string>
#include <vector>
using namespace std;
#define REP(i,a,n) for(int i=(a); i<(int)(n); i++)
#define rep(i,n) REP(i,0,n)
#define FOR(it,c) for(__typeof((c).begin()) it=(c).begin(); it!=(c).end(); ++it)
#define ALLOF(c) (c).begin(), (c).end()
typedef long long ll;

vector<ll> factorize(ll x){
  vector<ll> ret;
  ll d, q;
  while(x>=4 && x%2==0){ret.push_back(2); x/=2;}
  d = 3; q = x/d;
  while(q>=d){
    if(x%d==0){ ret.push_back(d); x=q; }else{ d+=2; }
    q=x/d;
  }
  ret.push_back(x);
  return ret;
}

map<ll,int> ret;

void dfs(ll now, const vector< pair<ll,int> >& v, int id){
  ret[now] = 1;
  if(id>=v.size()) return;

  dfs(now,v,id+1);
  ll p = now;
  rep(i,v[id].second){
    p *= v[id].first;
    dfs(p,v,id+1);
  }
}

int main(){
  ios::sync_with_stdio(false);
  ll N;
  cin >> N;
  vector<ll> a = factorize(N);
  vector<ll> b = factorize(N-1);
  map<ll,int> ma, mb, mab;
  rep(i,a.size()){
    ma[a[i]]++;
    mb[a[i]]++;
  }
  rep(i,b.size()){
    mb[b[i]]++;
  }
  mb[2]--;
  FOR(it,ma){
    if(mb.count(it->first)>0){
      mab[it->first] = min(it->second, mb[it->first]);
    }
  }

  vector< pair<ll,int> > v;
  FOR(it,mab){
    v.push_back(make_pair(it->first,it->second));
  }
  dfs(1LL,v,0);
  ll sum = 0;
  FOR(it,ret){
    sum += it->first;
  }
  cout << sum << endl;
  return 0;
}