Submission #307471

#	Time	Username	Problem	Language	Result	Execution time	Memory
307471		keta_tsimakuridze	Lottery (CEOI18_lot)	C++14	0 / 100	6 ms	1536 KiB

lot

#include<bits/stdc++.h>
#define f first
#define s second
using namespace std;
const int N = 1e4+5;
int n , l , q , a[N] , b[N] , cur , ans[105][N] , Ans[105][N] , aft[N];
pair< int , int > p[105];
int main() {
	cin>>n>>l;
	for(int k=1; k<=n; k++)
		cin>>a[k];
	cin>>q;
	for(int k=1; k<=q; k++) {
		cin>>p[k].f;
		p[k].s=k;
	}
	//k___l
	// k-ze meti an toli
	sort(p+1,p+q+1);
	cur=1;
	for(int k=0; k<=n; k++){
		while(p[cur].f<k && cur<=q) {
			cur++;
		}
		aft[k]=cur;
	//	cout<<k<<" "<<cur<<endl;
	}

	for(int d=1; d<=n; d++) {
		for(int k=1; k<=n-d; k++) {
			b[k]=0;
			if(a[k]!=a[k+d]){
				b[k]=1;
			}
		}
		//cout<<d<<"___"<<endl;
		for(int k=1; k<=n-d; k++){
			
		//	cout<<b[k]<<" ";
			b[k]+=b[k-1];
			if( k>=l ){
				int K=b[k]-b[k-l];
				ans[k-l+1][aft[K]]++;
				ans[k+d-l+1][aft[K]]++;
			}
		}
	}
	/*
	for(int k=1; k<=n; k++){
		for(int i=1; i<=n; i++){
			cout<<ans[i][k]<<" ";
		}
		cout<<endl;
	}
	*/
	for(int k=1; k<=q; k++){
		for(int i=1; i<=n-l+1; i++){
			Ans[i][p[k].s]=ans[i][k]+Ans[i][p[k-1].s];
		}
	}
	
	for(int k=1; k<=q; k++){
	
		for(int i=1; i<=n-l+1; i++){
			cout<<Ans[i][k]<<" ";
		}
		cout<<endl;
	}
}

• Subtask 1: 0 / 25
• Subtask 2: 0 / 20
• Subtask 3: 0 / 20
• Subtask 4: 0 / 15
• Subtask 5: 0 / 20