Codeforces 1371 E1. Asterism (Easy Version)

链接: E1. Asterism (Easy Version)

题意:
一个初始值 x 进行 n 次比较 如果 x>a[i],则 x++; 要使 x每次都大于 a[i] ,数组 a 有多少种不同的置换 ,f(x)即为不同的置换数。问有多少个 x 满足 f(x)%p !=0;
思路:

  1. 设 a数组的最大值 为 m ,如果 x >=m 随便怎么放都是可行的,所以置换数位 n!,而 n!%p一定等于 0,所以一定不满足。
  2. 而如果 x+n-1< m,最后的值达不到 m 是没有满足的置换的,0%p=0 ,所以也不满足。
  3. 所以只要枚举 x-n-1 到 m 求出 每个 f(x),看是否满足题意。复杂度 (n*n);
  4. 求 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 
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;
}

你可能感兴趣的:(Codeforces 1371 E1. Asterism (Easy Version))