結果

問題 No.2930 Larger Mex
コンテスト
ユーザー highlighter
提出日時 2024-10-12 14:58:30
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,640 bytes
コンパイル時間 5,661 ms
コンパイル使用メモリ 315,612 KB
実行使用メモリ 18,944 KB
最終ジャッジ日時 2024-10-12 14:58:55
合計ジャッジ時間 25,198 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 28 WA * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
#define int long long

signed main(){
	int N,M;
	cin >> N >> M;
	vector<int> A(N);
	for(int i=0;i<N;i++){
		cin >> A[i];
	}
	if(M==0){
		for(int k=1;k<=N;k++){
			cout << N-k+1 << endl;
		}
		return 0;
	}
	//k>=Mが必要．
	//長さkの連続部分列であって，0,1,2,...,M-1がすべて含まれる　に言い換えられる
	//lを移動 -> rの最小値が求まれば良い
	//尺取りのノリ
	map<int,int> data;
	int r=-1;
	for(int R=0;R<N;R++){
		if(A[R]<M){
			data[A[R]]++;
		}
		if((int)(data.size())==M){
			r=R+1;
			break;
		}
	}
	if((int)(data.size())!=M){
		for(int k=1;k<=N;k++){
			cout << 0 << endl;
		}
		return 0;
	}
	//[0,r)はok
	//長さr,r+1,...,Nに+1
	//imos
	//1,2,...,Nに+1
	vector<int> ans(N+1);
	ans[r-1]++;
	ans[N]--;
	for(int l=1;l<N;l++){
		//l-1を削除
		if(A[l-1]<M){
			data[A[l-1]]--;
			if(data[A[l-1]]==0){
				data.erase(A[l-1]);
			}
		}
		if((int)(data.size())==M){
			ans[r-l-1]++;
			ans[N-l]--;
			if(l==r){
				r++;
				data.clear();
				if(r<N){
					if(A[r]<M){
						data[A[r]]++;
					}
				}
				continue;
			}
			continue;
		}
		while(r<N){
			if(A[r]<M){
				data[A[r]]++;
			}
			r++;
			if((int)(data.size())==M){
				break;
			}
		}
		if((int)(data.size())==M){
			//[l,r) ~ [l,N)
			//r-l ~ N-lに+1
			ans[r-l-1]++;
			ans[N-l]--;
		}
		if(l==r){
			r++;
			if(r<N){
				if(A[r]<M){
					data[A[r]]++;
				}
			}
		}
	}
	vector<int> sum(N+1);
	sum[0]=0;
	for(int i=1;i<=N;i++){
		sum[i]=sum[i-1]+ans[i-1];
	}
	for(int i=1;i<=N;i++){
		cout << sum[i] << endl;
	}
	cout << endl;
}
0