結果
問題 |
No.944 煎っぞ!
|
ユーザー |
|
提出日時 | 2019-12-07 00:09:58 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 24 ms / 3,000 ms |
コード長 | 2,747 bytes |
コンパイル時間 | 2,793 ms |
コンパイル使用メモリ | 203,160 KB |
最終ジャッジ日時 | 2025-01-08 09:26:51 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 35 |
ソースコード
#include <bits/stdc++.h> #define REP(i, n) for (int i = 0; (i) < (int)(n); ++ (i)) #define REP3(i, m, n) for (int i = (m); (i) < (int)(n); ++ (i)) #define REP_R(i, n) for (int i = (int)(n) - 1; (i) >= 0; -- (i)) #define REP3R(i, m, n) for (int i = (int)(n) - 1; (i) >= (int)(m); -- (i)) #define ALL(x) std::begin(x), std::end(x) using namespace std; /** * @note O(\sqrt{n}) */ struct prepared_primes { int size; std::vector<int> sieve; std::vector<int> primes; prepared_primes(int size_) : size(size_) { sieve.resize(size); REP3 (p, 2, size) if (sieve[p] == 0) { primes.push_back(p); for (int k = p; k < size; k += p) { if (sieve[k] == 0) { sieve[k] = p; } } } } std::vector<int64_t> list_prime_factors(int64_t n) { assert (1 <= n and n < (int64_t)size * size); std::vector<int64_t> result; // trial division for large part for (int p : primes) { if (n < size or n < (int64_t)p * p) { break; } while (n % p == 0) { n /= p; result.push_back(p); } } // small part if (n == 1) { // nop } else if (n < size) { while (n != 1) { result.push_back(sieve[n]); n /= sieve[n]; } } else { result.push_back(n); } assert (std::is_sorted(ALL(result))); return result; } std::vector<int64_t> list_all_factors(int64_t n) { auto p = list_prime_factors(n); std::vector<int64_t> d; d.push_back(1); for (int l = 0; l < p.size(); ) { int r = l + 1; while (r < p.size() and p[r] == p[l]) ++ r; int n = d.size(); REP (k1, r - l) { REP (k2, n) { d.push_back(d[d.size() - n] * p[l]); } } l = r; } return d; } }; int solve(int n, const vector<int> & a) { int sum = accumulate(ALL(a), 0); prepared_primes primes(sqrt(sum) + 3); auto ds = primes.list_all_factors(sum); sort(ALL(ds)); for (int d : ds) { int acc = 0; REP (i, n) { acc += a[i]; if (acc > d) { break; } else if (acc == d) { acc = 0; } } if (acc == 0) { return sum / d; } } assert (false); } int main() { int n; cin >> n; vector<int> a(n); REP (i, n) { cin >> a[i]; } cout << solve(n, a) << endl; return 0; }