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

• 問題: No.278 連続する整数の和(2)
• ユーザー: airis
• 提出日時: 2015-09-05 16:26:02
• 言語: C++11 (gcc 11.4.0)
• 結果: WA (最新)
  AC
  (最初)
• 実行時間: -
• コード長: 1,637 bytes
• コンパイル時間: 726 ms
• コンパイル使用メモリ: 84,992 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-26 21:30:37
• 合計ジャッジ時間: 1,700 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	13 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 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	2 ms; 5,376 KB
testcase_11	AC	3 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	10 ms; 5,376 KB
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-

ソースコード

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <numeric>
#include <bitset>
#include <complex>
#define rep(x, to) for (int x = 0; x < (to); x++)
#define REP(x, a, to) for (int x = (a); x < (to); x++)
#define foreach(itr, x) for (typeof((x).begin()) itr = (x).begin(); itr != (x).end(); itr++)
#define EPS (1e-14)

using namespace std;

typedef long long ll;
typedef pair<int, int> PII;
typedef pair<ll, ll> PLL;
typedef complex<double> Complex;
typedef vector< vector<int> > Mat;

ll N;
ll ans = 1;

vector<PLL> prime_factorization(ll x) {
	vector<PLL> res;
	for (ll i = 2; i * i <= x; i++) {
		if (x % i != 0) continue;
		PLL cur = PLL(0, 0);
		cur.first = i;
		while (x % i == 0) {
			cur.second++;
			x /= i;
		}
		res.push_back(cur);
	}
	// xが素数であった場合はxを追加
	if (x != 1) {
		res.push_back(PLL(x, 1));
	}
	return res;
}

void solve(ll x) {
	vector<PLL> factor = prime_factorization(x);
#if 0
	rep(i, factor.size()) {
		cout << factor[i].first << " " << factor[i].second << endl;
	}
#endif

	for (int i = 0; i < factor.size() && factor[i].first != x; i++) {
		ll p = factor[i].first;
		ll size = factor[i].second;
		ll tmp = p;
		rep(j, size) {
			tmp *= p;
		}
		ans *= (tmp - 1) / (p - 1);
	}
	// 最後の素数が自身(x)であった場合
	if (factor.size() > 0 && factor.back().first == x) {
		ans *= (1 + factor.back().first);
	} 
	cout << ans << endl;
}

int main() {
	cin >> N;
	solve(N % 2 == 0 ? N / 2 : N);
	return 0;
}