[HAOI2012]容易题 题解

题目描述

为了使得大家高兴，小Q特意出个自认为的简单题（easy）来满足大家，这道简单题是描述如下：有一个数列A已知对于所有的A[i]都是1~n的自然数，并且知道对于一些A[i]不能取哪些值，我们定义一个数列的积为该数列所有元素的乘积，要求你求出所有可能的数列的积的和 mod 1000000007的值，是不是很简单呢？
输入输出格式
输入格式：
第一行三个整数n,m,k分别表示数列元素的取值范围，元素个数，和已知的限制条数。
接下来k行，每行两个正整数x,y表示A[x]的值不能是y。
输出格式：
一行一个整数表示所有可能的数列的积的和对1000000007取模后的结果。如果一个合法的数列都没有，答案输出0。
输入输出样例
输入样例#1：
3 4 5
1 1
1 1
2 2
2 3
4 3
输出样例#1：
90

分析

一道数学题，我们可以先计算没有限制的点，再计算有限制的位置，对于没有限制的位置，设是$kkk$，此时$ans=pow（n*（n+1）/2，kkk）$，而对于有每个限制的点，是$n*（n+1）/2-限制数$，然后相乘，就是答案。
上代码

#include<iostream>
#include<cmath>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int mod=1000000007;
const int maxn=100009;
ll sum[maxn];
struct data{
    ll x,y;
}a[maxn];
bool cmp(data a1,data a2){
    if(a1.x==a2.x)
        return a1.y<a2.y;
    else
        return a1.x<a2.x;
}
int poww(int x,int y,int mod){
    int ret=1;
    while(y){
        if(y&1)
            ret=1ll*ret*x%mod;
        x=1ll*x*x%mod;
        y>>=1;
    }
    return ret;
}
int main(){
    ll ans=0;
    int n,m,k;
    cin>>n>>m>>k;
    ll sgm=(1ll*n*(n+1)>>1)%mod;
    for(int i=1;i<=k;i++)
        cin>>a[i].x>>a[i].y;
    sort(a+1,a+k+1,cmp);
    int cnt=0;
    for(int i=1;i<=k;i++){
        if(a[i].x!=a[i-1].x)
            sum[++cnt]=sgm;
        else
            if(a[i].y==a[i-1].y)
                continue;
        sum[cnt]=(sum[cnt]-a[i].y+mod)%mod;
    }
    ans+=poww(sgm,m-cnt,mod);
    for(int i=1;i<=cnt;i++)
        ans=(ans*sum[i]+mod)%mod;
    cout<<ans<<endl;
    return 0;
}