CF Round#654(div.2) E1. Asterism (Easy Version)

题目介绍

链接: link.

中文描述

Yuzu有x个糖果。她的对手有n个(从1编号至n)，每个对手有ai个糖果。对手的不同排列为P，排列中第i个敌人为Pi。定义有以下的pk规则：

1. 如果Yuzu当前糖果数量大于等于第Pi个敌人的糖果数，她胜利并获得一个糖果；
2. 获得的糖果计入x依次和接下来第i+1个pk。

定义f(x)为初始糖果数量为x，最终能够获得胜利的对手排列种类数。

然后题目问：给出n,p（p是一个素数）和每个敌人的糖果数ai。有哪些x可以使得f(x)不被p整除。
其中(2≤p≤n≤2000)，(1≤ai≤2000)

题解

本题是一个模拟，乘法原理题目(思维题？)。
首先定义序列a[i]中最大值为Max，最小为Min。易知若x>=Max，则f(x)=n! 含p因子；而x<Min时f(x)=0也为p的倍数。故只需要在[Min，Max)内枚举x即可。
接着假设对1个初始的糖果数量x，对手糖果数量小于等于他的人数为v。则第一个位置有v种选择。接着下一次pk时当前有(x+1)枚糖果，糖果数小于等于(x+1)的敌人数量为：v+（糖果数为x+1）的敌人数量。又由于第一个位置排列时用掉一个，则第二个位置可填的种类数为：v+（糖果数为x+1）的敌人数量 - 1。第3个位置可排列种类数则为：v+（糖果数为x+1的敌人数量）+ （糖果数为x+2的敌人数量）- 2。……
以此类推，以乘法原理记录种类数f(x)即可。由于只关注f(x)是否为p的倍数，故只需要模拟过程中判断乘数是否为p。

代码

#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long ll;
const int maxn=4005;
int a[maxn];
int sum[maxn];
bool flag[maxn];

int main(){
	memset(flag,false,sizeof(flag));
	int N,p,x,Max=0,Min=2001; scanf("%d%d",&N,&p);
	for(int i=1;i<=N;i++) scanf("%d",&x),a[x]++,Max=max(Max,x),Min=min(Min,x);
	
//	[1,i]区间内糖果个数 
	for(int i=1;i<=Max;i++) sum[i]=sum[i-1]+a[i];
	int v,cnt=0;
	bool F;
	for(int x=Min;x<Max;x++){
//		printf("x=%d\n",x);
		F=true;
		v=sum[x-1]; 
//		simulate
		for(int i=x;i<x+N&&F;i++){
			v+=a[i];
//			printf("v=%d\n",v);
			if(v%p==0) F=false; 
			v--;
		} 
		if(F) flag[x]=true,cnt++;
	}
	printf("%d\n",cnt);
	for(int i=Min;i<Max;i++){
		if(flag[i]) printf("%d ",i);
	}
	printf("\n");
}