結果
問題 |
No.3257 +|+
|
ユーザー |
|
提出日時 | 2025-09-06 18:15:23 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,237 bytes |
コンパイル時間 | 3,025 ms |
コンパイル使用メモリ | 220,836 KB |
実行使用メモリ | 354,048 KB |
最終ジャッジ日時 | 2025-09-06 18:15:42 |
合計ジャッジ時間 | 18,766 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 6 WA * 27 |
ソースコード
#pragma GCC target("prefer-vector-width=512") #pragma GCC optimize("Ofast") #pragma GCC optimize("unroll-loops") #include <bits/stdc++.h> using namespace std; using ll = long long; #define rep(i, s, t) for (ll i = s; i < (ll)(t); i++) #define all(x) begin(x), end(x) template <class T> bool chmin(T& x, T y) { return x > y ? (x = y, true) : false; } template <class T> bool chmax(T& x, T y) { return x < y ? (x = y, true) : false; } struct io_setup { io_setup() { ios::sync_with_stdio(false); cin.tie(nullptr); cout << fixed << setprecision(15); } } io_setup; void solve() { int n; cin >> n; vector<int> a(n); rep(i, 0, n) cin >> a[i]; constexpr int sq = 448; int sqn = min(n, sq); ll ans = 0; rep(i, 0, sqn) rep(j, i + 1, sqn) { if ((a[i] + a[j]) % (i + j + 2) == 0) ans++; } constexpr int shf = 2e5; array<array<uint16_t, shf * 2 + 1>, sq> mp = {}; rep(i, 0, sqn) { ll tmp = i + 1 - a[i]; rep(k, 0, sqn) { mp[k][shf + tmp]++; tmp += i + 1; } } rep(i, sqn, n) { ll tmp = i + 1 - a[i]; rep(k, 0, sqn) { if (tmp > shf) break; ans += mp[k][shf - tmp]; mp[k][shf + tmp]++; tmp += i + 1; } } cout << ans << "\n"; } int main() { int t = 1; // cin >> t; while (t--) solve(); }