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

• 問題: No.278 連続する整数の和(2)
• ユーザー: airis
• 提出日時: 2015-09-05 16:19:54
• 言語: C++11 (gcc 11.4.0)
• 結果: WA
• 実行時間: -
• コード長: 1,572 bytes
• コンパイル時間: 752 ms
• コンパイル使用メモリ: 84,712 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-19 03:49:44
• 合計ジャッジ時間: 1,497 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
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	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	AC	2 ms; 5,376 KB
testcase_13	WA	-
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 < (int)factor.size() - 1; i++) {
		ll p = factor[i].first;
		ll size = factor[i].second;
		ll tmp = p;
		rep(j, size) {
			tmp *= p;
		}
		ans *= (tmp - 1) / (p - 1);
	}
	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;
}