2751: [HAOI2012]容易题(easy)

题目链接

题目大意：一个数列A已知对于所有的A[i]都是1~n的自然数，并且知道对于一些A[i]不能取哪些值，我们定义一个数列的积为该数列所有元素的乘积，求出所有可能的数列的积的和

题解：用小数据手玩一下公式，设m=2，第一项可行的取值为$a_1,a_2$，第二项可行的取值为$b_1,b_2$，有

$ans=a_1b_1+a_1b_2+a_2b_1+a_2b_2=(a_1+a_2)*(b_1+b_2)$

然后拓展一下，发现答案等于各项可取的数之和的积

由于k很小，有限制的项占少数，暴力算有限制的即可
再把答案乘上$({n*(n+1)\over 2})^{num}$，num为无限制的项数

还要去一下重

我的收获：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;

#define M 100005
#define P 1000000007
#define pii pair<int,int>
typedef long long ll;

int n,m,k,x,y,cnt;
long long tmp,ans=1,a[M];
pii u[M];

ll fpow(ll a,int b){
    ll c=1;a%=P;
    for(;b;b>>=1,a=(a*a)%P)
    if(b&1) c=(c*a)%P;
    return c%P;
}

void work()
{
    ll ret=0;
    for(int i=1;i<=k;i++){
        ret+=u[i].second;ret%=P;
        if(i==k||u[i].first!=u[i+1].first){
            ans=ans*((tmp-ret+P)%P)%P;
            ret=0;cnt++;
        }
    }
    ans=ans*fpow(tmp,m-cnt)%P;
    cout<<ans<<endl;
}

void init()
{
    cin>>n>>m>>k;
    for(int i=1;i<=k;i++){
        scanf("%d%d",&x,&y);
        u[i]=make_pair(x,y);
    }
    sort(u+1,u+1+k);
    k=unique(u+1,u+1+k)-(u+1);
    tmp=(ll)n*(n+1)/2%P;
}

int main()
{
    init();
    work();
    return 0;
}