Submission #599766

#TimeUsernameProblemLanguageResultExecution timeMemory
599766CSQ31Lottery (CEOI18_lot)C++17
80 / 100
3060 ms1364 KiB
#include <bits/stdc++.h>
using namespace std;
#define owo ios_base::sync_with_stdio(0);cin.tie(0);
typedef long long int ll;
#define all(a) a.begin(),a.end()
int a[10000],b[10000];
int ans[100][10000];
int main()
{
	owo
	int n,l;
	cin>>n>>l;
	for(int i=0;i<n;i++)cin>>a[i];
	int q;
	cin>>q;
	vector<array<int,2>>qu;
	for(int i=0;i<q;i++){
		int k;
		cin>>k;
		qu.push_back({l-k,i});
		//>=l-k matches
	}
	sort(all(qu));
	for(int i=-n+1;i<n;i++){
		if(i==0)continue;
		for(int j=0;j<n;j++){
			if(j+i>=0 && j+i<n && a[j] == a[i+j])b[j] = 1;	
		}
		for(int j=1;j<n;j++)b[j]+=b[j-1];
		for(int j=max(0,-i);j<min(n-l+1,n-l-i+1);j++){ //i am shiftng by i
			int num = b[j+l-1] - (j?b[j-1]:0);
			for(int k=0;k<q;k++){
				if(num >= qu[k][0])ans[qu[k][1]][j]++;
			}
		}
		for(int j=0;j<n;j++)b[j] = 0;
	}
	for(int i=0;i<q;i++){
		for(int j=0;j<n-l+1;j++)cout<<ans[i][j]<<" ";
		cout<<'\n';
	}
	
	
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...