E1. Asterism (Easy Version) (暴力,二分查找)

    技术2022-07-20  65

    题目传送门

    题意: 假设你有x颗糖果,你面前有n个敌人,第i个敌人有a[i]颗糖果,你可以选择对战的顺序,如果你手里的糖果不比敌人少,那么你胜利并获得一颗糖果,现在我们定义f(x)为你手里有x颗糖果时,你能打败所有对手的方案数,对于给定质数p,如果f(x)与p互质,那么这个x就是符合要求的,现在给定n,p和a[i],问你有多少个x符合条件,并且输出这些x。

    数据范围: 2<=p<=n<=2000,a[i]<=2000

    思路: 显然,如果x比2000要大的话,如果这个x合法,那么所有大于x的都合法了,这显然就有无限个x合法了,所以可以把x的范围控制在2000以内,那么我们先对a[i]排序,在2000以内枚举x,用cnt表示已经战胜的敌人,我们每次就二分查找a数组中小于x+cnt的数量num,num-cnt就是目前没有交战的敌人数量,那么方案数是要乘num-cnt的,所以如果num-cnt是p的倍数,就可以直接break了,至于f(x)是多少,我们并不care,(如果一直乘,最后对p取模,c++的long long可能存不下而且时间复杂度也高)。最后只需要判断是否合法并存储答案就行了。

    #include<bits/stdc++.h> #pragma GCC optimize("Ofast") #define endl '\n' #define null NULL #define ls p<<1 #define rs p<<1|1 #define fi first #define se second #define mp make_pair #define pb push_back #define ll long long #define int long long #define pii pair<int,int> #define ull unsigned long long #define all(x) x.begin(),x.end() #define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0); #define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n"; char *fs,*ft,buf[1<<20]; #define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++; inline int read(){int x=0,f=1;char ch=gc(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();} return x*f;} using namespace std; const int N=2e5+5; const int inf=0x3f3f3f3f; const int mod=1e9+7; const double eps=1e-7; const double PI=acos(-1); int a[N]; signed main() { int n,p; cin>>n>>p; for(int i=1;i<=n;i++) cin>>a[i]; sort(a+1,a+n+1); vector<int>v; for(int i=1;i<=2000;i++) { int cnt=0,res=0,fg=1; for(int j=1;j<=n;j++) { int pos=upper_bound(a+1,a+n+1,i+cnt)-a; pos--; if((pos-cnt)%p==0) { fg=0; break; } cnt++; } if(fg) v.pb(i); } cout<<v.size()<<endl; for(auto i:v) cout<<i<<' '; }
    Processed: 0.013, SQL: 9