Description
小A得到了一个数列A,他告诉你这个数列中所有的A[i]都是1到n的自然数,并且告诉你对于一些A[i]不能取哪些值。无聊的你想要知道所有可能的数列的积的和。定义一个数列的积为这个数列所有数的乘机。由于答案太大,只要模10^9+7输出即可。
Input
第一行三个整数n,m,k,分别表示数列元素取值范围,数列长度以及限制的数量。
接下来k行每行两个正整数x,y,表示A[x]的值不可能是y。
Output
一行一个整数表示答案。如果连一种可能的数列都没有,就输出0即可。
Hint
对于前30%的数据,n<=4,m<=10,k<=10
另有20%的数据k=0
对于70%的数据n<=1000,m<=1000,k<=1000
对于100%的数据,n<=10^9,m<=10^9,k<=10^5,1<=y<=n,1<=x<=m
Source
BY BPM
Solution
设c[i]表示为某种可能的方案前i位的乘积,那么不难得到在无限制的情况下
sum[i]=∑c[i−1]∗n(n+1)2
,即
sum[i]=sum[i−1]∗n(n+1)2,sum[0]=1
这时候加上限制就比较容易了,求和的时候减去就行。
观察数据发现n很大,k很小。最坏情况下最多k个限制,剩下n-k个都是没有限制、 n(n+1)2 的乘积。相同的部分快速幂,有限制的部分单独乘起来就可以了。
Code
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i, st, ed) for (int i = st; i <= ed; i += 1)
#define fill(x, t) memset(x, t, sizeof(x))
#define ll long long
#define MOD 1000000007
#define N 100001
struct data{ll x, y;}v[N];
int rk[N]; ll c[N];
inline bool cmp(data a, data b){return a.x < b.x || a.x == b.x && a.y < b.y;}
inline ll pow(int dep, int x){
if (dep == 0){return 1;}
if (dep == 1){return x;}
ll tmp = pow(dep / 2, x);
return (dep&1)?((tmp%MOD*tmp%MOD*x%MOD)%MOD):((tmp%MOD*tmp% MOD)%MOD);
}
inline ll read(){
ll x = 0; char ch = getchar();
while (ch < '0' || ch > '9'){ch = getchar();}
while (ch <= '9' && ch >= '0'){x = x * 10 + ch - '0'; ch = getchar();}
return x;
}
int main(void){
freopen("math.in", "r", stdin);
freopen("math.out", "w", stdout);
ll n = read(), m = read(); int k = read();
ll lim = ((n+1)*n/2)%MOD;
rep(i, 1, k){
ll x = read(), y = read();
v[i] = (data){x, y};
}
std:: sort(v + 1, v + k + 1, cmp);
rk[1] = 1;
rep(i, 2, k){
if (v[i].x == v[i - 1].x){rk[i] = rk[i - 1];}
else{rk[i] = rk[i - 1] + 1;}
if (v[i].x == v[i - 1].x && v[i].y == v[i - 1].y){
v[i - 1].x = v[i - 1].y = 0;
}
}
rep(i, 0, rk[k]){c[i] = 0;}
rep(i, 1, k){
if (v[i].x){c[rk[i]] = (c[rk[i]] + v[i].y) % MOD;}
}
ll ans = 1;
rep(i, 1, rk[k]){ans = (ans * (lim + MOD - c[i])) % MOD;}
ans = (ans%MOD*pow(m - rk[k], lim))%MOD;
printf("%lld\n", ans);
return 0;
}