CF1070K 出现 Unknown Error?
#include<bits/stdc++.h>
using namespace std;
int n,k,a[1000001],sum,num,ans[10000001],nn;
int main()
{
cin>>n>>k;
nn=k;
for(int i = 1;i<=n;i++)
{
cin>>a[i];
sum+=a[i];
}
num=sum/k;
if(sum%k!=0)
{
cout<<"No"<<endl;
return 0;
}
int ji=0;
int jsj=0;
while(k>=1)
{
k--;
int b = 0;
int js = 1;
while(b+a[js+jsj]<num)
{
b+=a[js+jsj];
js++;
}
b+=a[js+jsj];
if(b!=num)
{
cout<<"No"<<endl;
return 0;
}
ans[++ji]=js;
jsj+=js;
}
cout<<"Yes"<<endl;
for(int i = 1;i<=nn;i++)
{
cout<<ans[i];
if(i!=nn)cout<<' ';
}
cout<<endl;
return 0;
}