无意写的,结果第二天加了几个访量,来加点细节吧
easy版本直接就是方案数暴力,好好想想再看hard吧
#include<iostream>
#include<algorithm>
using namespace std;
const int N=2010;
int a[N];
int xl[N];
int main()
{
int n,p;
cin>>n>>p;
for(int i=1;i<=n;i++) scanf("%d",a+i);
sort(a+1,a+n+1);
for(int i=1;i<=2000;i++)
{
int last=1;
bool flag=1;
for(int j=1;j<=n;j++)
{
while(last+1<=n&&a[last+1]<=j+i-1) last++;
if(a[last]>j+i-1||last<j) flag=0;
if((last-j+1)%p==0) flag=0;
if(!flag) break;
}
if(flag) xl[++xl[0]]=i;
}
cout<<xl[0]<<endl;
for(int i=1;i<=xl[0];i++) cout<<xl[i]<<' ';
}
我们发现糖果的数量从前向后是公差为1的等差数列
cnt数组存放右边所能打死的敌人个数
当初始糖果加一的时候整个序列的cnt会左移整体加一,所以可以很方便的查询到是否存在p因子
#include<iostream>
#include<map>
#include<algorithm>
using namespace std;
const int N=1e5+10;
int a[N];
int xl[N];
map<int,int> mp;
int cnt[N];
int main()
{
int n,p;
cin>>n>>p;
for(int i=1;i<=n;i++) scanf("%d",a+i);
sort(a+1,a+n+1);
int mx=0;
int top;
for(int i=1;i<=n;i++) mx=max(mx,a[i]-i);
top=mx+1;
for(int i=1,last=1;i<=n;i++)
{
while(last<n&&a[last+1]<=i+top-1) last++;
cnt[i]=last-i+1;
mp[cnt[i]%p]++;
}
for(int i=0;i<n;i++)
{
if(!mp[(p-i%p)%p]) xl[++xl[0]]=top+i;
mp[cnt[i+1]%p]--;
mp[(-i%p+p)%p]++;
}
cout<<xl[0]<<endl;
for(int i=1;i<=xl[0];i++) printf("%d ",xl[i]);
puts("");
}