vfk 式名字重排 【排列+逆元】

题目传送门

题目描述

传说中很多大牛喜欢用自己姓名拼音的重排列，比如著名的吕凯风大牛，他的姓名拼音是lvkaifeng，重新排列之后就得到了VFleaKing，也就是著名的伏特跳蚤国王。现在，火星人觉得VFleaKing太强啦！于是他们也要这样取他们的ID名，现在问题来了，他们的名字都非常长，而且都是由火星文组成的（火星文有好多好多字符）。火星人想请你帮他们算算如果把他们的名字重排有几种方案，当然了，原序列也是一种方案啦~如果你不帮他们算出正确的答案你就会死哦~

形式化描述：给定一个长度为N的整数序列A1…N，求有多少种不同长度为的N的整数序列B1…N是A的重排，即可重集{Ai}={Bi}。两个序列不同当且仅当它们任一位置上的元素不相等。

输入

第1行：两个正整数N,k，其中N代表该火星人名字的长度，k代表火星文有多少种字符，我们不妨设在有k种字符的情况下的火星文中的字符分别是0到k−1。
第2行：用空格隔开的N个正整数，代表这个火星人的名字，我们保证这N个数字一定在0到k−1的范围内。

输出

第1行：一个正整数——这个火星人的名字重排方案数对109+7取模的结果（火星人的逻辑特别奇怪，只要知道模数是多少就可以放过你了）。

样例输入

复制样例数据

4 2
0 0 1 1

样例输出

6

提示

满足条件的重排共有如下6种：
0,0,1,1
0,1,0,1
1,0,0,1
0,1,1,0
1,0,1,0
1,1,0,0

解题思路：求出每个数出现的次数num[i]

ans=(n!/(num[0]!*num[1]*...*num[k]!)%mod,为了避免超时，先打一个排列的表。

除法不能直接取模想出，所以要求逆元（重点）。

AC代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
using namespace std;
#define io ios::sync_with_stdio(0),cin.tie(0)
#define ms(arr) memset(arr,0,sizeof(arr))
#define mc(a,b) memcpy(a,b,sizeof(b))
#define inf 0x3f3f3f
#define fin freopen("in.txt", "r", stdin)
#define fout freopen("out.txt", "w", stdout)
typedef long long ll;
typedef unsigned long long ULL;
const int mod=1e9+7;
const int N=2e6+7;
int n,k;
int num[N];

ll a[N];

void get_a()
{
    a[0]=1;
    for(int i=1;i<=2e6;i++){
        a[i]=(i*a[i-1])%mod;
    }
}

ll qpow(ll a,ll b,ll p)
{
    ll tmp = 1;
    while(b)
    {
        if(1&b) tmp = (tmp*a)%p;
        a = (a*a)%p;
        b>>=1;
    }
    return tmp;
}
ll inv(ll a,ll p) //费马小定理求逆元
{
    return qpow(a,p-2,p);
}

int main()
{
//    fin;
    scanf("%d%d",&n,&k);
    int x;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&x);
        num[x]++;
    }
    get_a();
    ll ans=a[n];
    for(int i=0;i<k;i++){
        if(num[i]>1){
//            ll m=inv(a[num[i]],mod-2);
//            cout<<m<<endl;
            ans=(ans*inv(a[num[i]],mod))%mod;
        }
    }
    printf("%lld\n",ans);
    return 0;
}