結果
問題 |
No.1079 まお
|
ユーザー |
|
提出日時 | 2020-07-12 18:52:47 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,210 bytes |
コンパイル時間 | 2,327 ms |
コンパイル使用メモリ | 209,008 KB |
最終ジャッジ日時 | 2025-01-11 20:04:38 |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 17 WA * 13 |
ソースコード
#include <bits/stdc++.h> using namespace std; int main() { /* ios::sync_with_stdio(false); */ int N, K; cin >> N >> K; vector<int> A(N); for (int i = 0; i < N; ++i) cin >> A[i]; function<int64_t(int, int)> dvcq = [&](int l, int r) { if (r - l < 1) return (int64_t) 0; int64_t res = 0; map<int, int> cnt; for (int i = l; i < r; ++i) ++cnt[A[i]]; if (cnt.begin()->second < 2) { map<int, int64_t> sum; for (int i = l; i < r; ++i) sum[A[i]] += i + 1; for (int i = l; i < r; ++i) { if (A[i] <= K) res += sum[K - A[i]] - (int64_t) i * cnt[K - A[i]]; --cnt[A[i]]; sum[A[i]] -= i + 1; } } else { int i = l; int j = i; while (j < r && A[j] != cnt.begin()->first) ++j; for (++j; j < r && A[j] != cnt.begin()->first; ++j) ; res += dvcq(i, j); while (true) { while (i < r && A[i] != cnt.begin()->first) ++i; if (++i >= r) break; res -= dvcq(i, j); for (++j; j < r && A[j] != cnt.begin()->first; ++j) ; if (j > r) j = r; res += dvcq(i, j); if (j == r) break; } } return res; }; cout << dvcq(0, N) << endl; return 0; }