洛谷-2220 [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

100%的数据 $n<=10^9,m<=10^9,k<=10^5,1<=y<=n,1<=x<=m$

解释：很容易推出公式,$dp[i]=\frac{(1+n{)}^{i-j}\ast n^{i-j}}{2^{i-j}}\ast dp[j]$,对于限制，然后我们就可以枚举限制来计算，对于限制，我们只需要套减去限制的数字*前一个状态就好了。

#include<iostream>
#include<algorithm>
#define mod 1000000007
#define ll long long
using namespace std;
struct node{
    int pos,v;
}s[100005];
ll pow(ll a,ll b){
    ll ret=1;
    while(b){
        if(b&1) ret=ret*a%mod;
        a=a*a%mod;b>>=1;
    }
    return ret;
}
ll inv(ll x){
    return pow(x,mod-2);
}
bool cmp(node &a,node &b){
    if(a.pos==b.pos) return a.v<b.v;
    return a.pos<b.pos;
}
int n,m,k;
ll ret=1;
ll frontret=0;
int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m>>k;
    for(int i=1;i<=k;i++){
        cin>>s[i].pos>>s[i].v;
    }
    k++;
    s[k].pos=m;s[k].v=0;
    sort(s+1,s+1+k,cmp);
    for(int i=1;i<=k;i++){
        if(s[i].pos!=s[i-1].pos){
            frontret=ret*pow(1+n,s[i].pos-s[i-1].pos-1)%mod*pow(n,s[i].pos-s[i-1].pos-1)%mod*inv(pow(2,s[i].pos-s[i-1].pos-1))%mod;
            ret=ret*pow(1+n,s[i].pos-s[i-1].pos)%mod*pow(n,s[i].pos-s[i-1].pos)%mod*inv(pow(2,s[i].pos-s[i-1].pos))%mod;
        }
        if(s[i].pos==s[i-1].pos&&s[i].v==s[i-1].v) continue;
        ret-=s[i].v*frontret;
        ret%=mod;ret+=mod;ret%=mod;
    }
    cout<<ret<<endl;
    return 0;
}