E1. Asterism(easy&hard)

无意写的，结果第二天加了几个访量，来加点细节吧
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("");
}