結果
問題 |
No.3257 +|+
|
ユーザー |
|
提出日時 | 2025-09-06 00:20:26 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,409 bytes |
コンパイル時間 | 1,181 ms |
コンパイル使用メモリ | 108,848 KB |
実行使用メモリ | 13,676 KB |
最終ジャッジ日時 | 2025-09-06 00:20:42 |
合計ジャッジ時間 | 14,224 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 8 TLE * 1 -- * 24 |
ソースコード
#include <iostream> #include <vector> #include <numeric> #include <unordered_map> #include <cmath> #include <algorithm> void solve() { int N; std::cin >> N; std::vector<int> A(N); for (int i = 0; i < N; ++i) { std::cin >> A[i]; } long long ans = 0; const int acm = 200000; int B = static_cast<int>(cbrt(2.0 * acm * acm / std::max(1, N))); if (B == 0) B = 1; for (int k = 1; k <= B; ++k) { std::unordered_map<long long, int> freq; for (int i = N; i >= 1; --i) { long long val_i = (long long)A[i - 1] - (long long)k * i; long long target = -val_i; if (freq.count(target)) { ans += freq[target]; } freq[val_i]++; } } const int S_limit = (acm + acm) / (B + 1); for (int i = 1; i <= N; ++i) { if (2 * i + 1 > S_limit) { break; } for (int j = i + 1; j <= N; ++j) { int s = i + j; if (s > S_limit) { break; } long long sum_A = (long long)A[i - 1] + A[j - 1]; if (sum_A % s == 0) { if (sum_A / s > B) { ans++; } } } } std::cout << ans << std::endl; } int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(0); solve(); return 0; }