1. 程式人生 > >“東信杯”廣西大學第一屆程式設計競賽(同步賽)B 不吉利的數(組合數+逆元)

“東信杯”廣西大學第一屆程式設計競賽(同步賽)B 不吉利的數(組合數+逆元)

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);
    }
}