链接: E1. Asterism (Easy Version)
题意:
一个初始值 x 进行 n 次比较 如果 x>a[i],则 x++; 要使 x每次都大于 a[i] ,数组 a 有多少种不同的置换 ,f(x)即为不同的置换数。问有多少个 x 满足 f(x)%p !=0;
思路:
- 设 a数组的最大值 为 m ,如果 x >=m 随便怎么放都是可行的,所以置换数位 n!,而 n!%p一定等于 0,所以一定不满足。
- 而如果 x+n-1< m,最后的值达不到 m 是没有满足的置换的,0%p=0 ,所以也不满足。
- 所以只要枚举 x-n-1 到 m 求出 每个 f(x),看是否满足题意。复杂度 (n*n);
- 求 f(x) ,总共有 n 个位置 从 m-n+1 到 m ,对应的值分别为 x到 m 。所以贪心的从大的开始安排,每个 a[i]只能安排到对应值比a[i]大或者等于的位置 。把最大的安排到可以满足的位置 一共 x+n-a[i] 个 第二大的数有 x+n-1-a[i]个位置。当 a[i]<=x 可以随便放。最后看所有可能 是否可以整除 p。
代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn=2e3+7;
typedef long long ll;
int T,n,p;
int a[maxn];
vector < int > ans;
int f(int x){
int sum=n,ans=1;
for(int i=1;i<=n;i++){
if(a[i]<=x){
ans=(ans*sum)%p;
}
else if(x+sum-a[i]<=0) return 0;
else ans=(ans*(x+sum-a[i]))%p;
sum--;
}
if(ans==0) return 0;
return 1;
}
int main(){
cin>>n>>p;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
sort(a+1,a+n+1);
reverse(a+1,a+n+1);
int m=a[1];
for(int i=max(1,m-n+1);i<m;i++){
if(f(i)) ans.push_back(i);
}
cout<<ans.size()<<endl;
for(auto i: ans) cout<<i<<" ";
cout<<endl;
}