結果
問題 | No.2930 Larger Mex |
ユーザー |
|
提出日時 | 2024-10-12 16:54:59 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 462 ms / 2,000 ms |
コード長 | 1,649 bytes |
コンパイル時間 | 2,280 ms |
コンパイル使用メモリ | 199,896 KB |
最終ジャッジ日時 | 2025-02-24 18:45:49 |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 50 |
ソースコード
#include <bits/stdc++.h> using namespace std; #include <atcoder/fenwicktree> using namespace atcoder; int main() { int N, M; cin >> N >> M; vector<int> A(N); for (int i = 0; i < N; i++) { cin >> A[i]; } vector<vector<int>> v(M); for (int i = 0; i < N; i++) { if (A[i] < M) { v[A[i]].push_back(i); } } fenwick_tree<int> ft(N); for (int i = 0; i < M; i++) { if (v[i].empty()) { continue; } ft.add(v[i][0], 1); } vector<int> imos(N + 2, 0); for (int i = 0; i < N; i++) { int le = i + 1, ri = N; int mn = N + 1; while (le <= ri) { int mid = le + (ri - le) / 2; if (ft.sum(i, mid) == M) { ri = mid - 1; mn = min(mn, mid); } else { le = mid + 1; } } if (mn != N + 1) { imos[mn - i]++; imos[N - i+ 1]--; } #ifdef hahna for (int j = 0; j < N; j++) { cout << ft.sum(j, j + 1) << " \n"[j + 1 == N]; } cout << "###" << mn << endl; #endif if (A[i] >= M) continue; auto ub = upper_bound(v[A[i]].begin(), v[A[i]].end(), i); if (ub != v[A[i]].end()) { ft.add(*ub, 1); } } for (int i = 1; i <= N; i++) { imos[i + 1] += imos[i]; } for (int i = 1; i <= N; i++) { cout << imos[i] << endl; } // fenwick tree de l ikou ni hajimete ataiga deta basyowo kanri // l wo hidari kara miteitte waga m ni naru tokoro wo nibutan de return 0; }