“東信杯”廣西大學第一屆程式設計競賽(同步賽)B 不吉利的數(組合數+逆元)
阿新 • • 發佈:2018-12-02
https://ac.nowcoder.com/acm/contest/283/B
簡單組合數學:
選出一共有多少個重複的 ,不同的
#include<bits/stdc++.h> using namespace std; typedef long long LL; const int N=2e5+5; const int M=1e9+7; LL a[N],F[N],Finv[N]; LL pow_(LL a,LL b){ a%=M; LL ret=1LL; while(b){ if(b&1) ret=ret*a%M; a=a*a%M; b>>=1; }return ret; } LL inv(LL a){return pow_(a,M-2) ;} void init(){//F是階乘,Finv是逆元的階乘 F[0]=Finv[0]=1LL; for(LL i=1;i<N;i++){ F[i]=F[i-1]*1LL*i%M; Finv[i]=Finv[i-1]*1LL*inv(i)%M; } } LL C(LL n,LL m) {//comb(n, m)就是C(n, m) if(m<0||m>n) return 0; return F[n]*1LL*Finv[n-m]%M*Finv[m]%M; } int main() { LL n,m; init(); while(~scanf("%lld %lld",&n,&m)){ for(int i=0;i<n;i++) scanf("%lld",&a[i]); LL c=unique(a,a+n)-a,b=n-c;//不同的 相同的; LL ans=C(c,m)+C(c-1,m-1)*b; printf("%lld\n",ans); } }