【容斥】51Nod 1829 函数

题面在这里

其实很简单……我太菜了……想了好久……

因为题目要求B中所有元素都至少被覆盖一次

所以枚举有几个元素没有被覆盖（这是容斥的基本套路）

答案就是：

$$\sum_{i=0}^m C_m^i\cdot (m-i)^n\cdot (-1)^{i+1}$$

示例程序：

#include<cstdio>
typedef long long ll;

const int maxn=1000005,tt=1e9+7;
int n,m;
ll fac[maxn],inv[maxn];
ll power(int a,int b){
    ll w=a,res=1;
    while (b){
        if (b&1) res=res*w%tt;
        w=w*w%tt;
        b>>=1;
    }
    return res;
}
void make(){
    fac[0]=inv[0]=1;
    for (int i=1;i<=n;i++) fac[i]=fac[i-1]*i%tt,inv[i]=power(fac[i],tt-2);
}
inline ll C(int y,int x){
    return fac[x]*inv[y]%tt*inv[x-y]%tt;
}
int main(){
    scanf("%d%d",&n,&m);
    if (n<m) return printf("0"),0;
    make();ll ans=0;
    for (int i=0;i<=m;i++)
     (ans+=power(m-i,n)*C(i,m)*((i&1)?-1:1))%=tt;
    printf("%lld",(ans+tt)%tt);
    return 0;
}