WA+TLE求调
查看原帖
WA+TLE求调
251074
学习柯学楼主2023/4/5 18:26
#include<bits/stdc++.h>
using namespace std;
long long n,k,a[5005],f[5005],i, sum, mmin=0x3f3f3f3f, ans;
int main() {
	cin>>n>>k;
	if(k==0) {
		for(i=1; i<=n; i++) {
			cin>>a[i];
			if(a[i]!=0)
				sum++;
		}
		cout<<sum;
	} else {
		for(i=1; i<=n; i++) {
			cin>>a[i];
			if(a[i]==k) {
				if(sum==k-1) {
					ans+=mmin;
				}
				mmin=0x3f3f3f3f;
				sum=0;
				memset(f,0,sizeof(f));
				continue;
			}
			if(a[i]<k&&f[a[i]]==0)
			{
				sum++;
			}
			f[a[i]]++;
		}
		if(sum==k) {
			for(i=1; i<k; i++)
				mmin=min(mmin,f[i]);
			ans+=mmin;
		}
		cout<<ans;
	}
	return 0;
}

2023/4/5 18:26
加载中...