結果

問題 No.1526 Sum of Mex 2
ユーザー spipipike
提出日時 2021-06-01 22:03:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 63 ms / 3,000 ms
コード長 1,147 bytes
コンパイル時間 2,027 ms
コンパイル使用メモリ 180,924 KB
実行使用メモリ 12,968 KB
最終ジャッジ日時 2024-11-09 00:18:38
合計ジャッジ時間 4,401 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for (int i = 0; i < (n); i++)
using namespace std;
using ll = long long;
using P = pair<int,int>;

int main(){
	int n;
	cin>>n;

	vector<vector<int>> ai(n+2);
	for(int i=1; i<=n+1; i++) ai[i].push_back(-1);
	rep(i, n){
		int a;
		cin>>a;
		ai[a].push_back(i);
	}
	for(int i=1; i<=n+1; i++) ai[i].push_back(n);

	ll tot=0;
	map<int, int> lr, rl;
	for(int i=1; i<=n+1; i++){
		if(i==1){
			auto calc=[](int a){ return (ll)(a+1)*a/2; };
			rep(j, ai[i].size()-1) tot+=calc(ai[i][j+1]-ai[i][j]-1);

			for(int j=1; j<ai[i].size()-1; j++){
				lr[ai[i][j]]=ai[i][j];
			}
		}
		else{
			rep(j, ai[i].size()-1){
				int l=ai[i][j], r=ai[i][j+1];
				
				int cl=l;
				vector<map<int, int>::iterator> ml;
				for(auto it=lr.lower_bound(l); it!=lr.end() && (it->second)<r; it++){
					tot+=(ll)((it->first)-cl)*(r-(it->second))*i;
					cl=(it->first);
					ml.push_back(it);
				}
				if(ml.size()>=1){
					int firstr=ml[0]->second, lastl=ml.back()->first;
					for(auto it : ml) lr.erase(it);
					if(1<=j) lr[l]=firstr;
					if(j<ai[i].size()-1) lr[lastl]=r;
				}
			}
		}
	}

	cout<<tot<<endl;
	return 0;
}
0