結果
問題 | No.956 Number of Unbalanced |
ユーザー |
![]() |
提出日時 | 2019-12-20 01:12:37 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 46 ms / 2,000 ms |
コード長 | 1,153 bytes |
コンパイル時間 | 2,386 ms |
コンパイル使用メモリ | 210,172 KB |
最終ジャッジ日時 | 2025-01-08 13:10:26 |
ジャッジサーバーID (参考情報) |
judge1 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 6 |
other | AC * 24 |
ソースコード
#include<bits/stdc++.h> using namespace std; using P = pair<int, int>; const int M = 1000000007; int main() { cin.tie(0); ios::sync_with_stdio(false); int n; cin >> n; vector<vector<int>> vv(100010); for (int i = 0; i < n; ++i) { int a; cin >> a; vv[a].push_back(i + 1); } long long ans = 0; for (const auto& v : vv) { if (v.size() == 0) continue; stack<P> st; st.push({0, 0}); unordered_map<int, long long> mp; int pd = 0, pj = 0, md = 0; for (int j : v) { int d = pd - j + pj + 2; md = min(md, d); if (pj < j - 1) { ans -= mp[pd]; if (mp.count(d - 1)) ans += mp[d - 1]; } while (!st.empty() && st.top().first < d) st.pop(); mp[d] += st.empty() ? j : j - st.top().second - st.top().first + d - 1; ans += mp[d] * 2; st.push({d, j}); pd = d; pj = j; } ans -= mp[pd]; for (int i = md; i < pd - n + pj; ++i) ans -= mp[i]; } cout << ans << '\n'; return 0; }