问题描述
小A得到了一个数列A,他告诉你这个数列中所有的A[i]都是1到n的自然数,并且告诉你对于一些A[i]不能取哪些值。无聊的你想要知道所有可能的数列的积的和。定义一个数列的积为这个数列所有数的乘机。由于答案太大,只要模10^9+7输出即可。
输入
第一行三个整数n,m,k,分别表示数列元素取值范围,数列长度以及限制的数量。
接下来k行每行两个正整数x,y,表示A[x]的值不可能是y。
输出
一行一个整数表示答案。如果连一种可能的数列都没有,就输出0即可。
样例输入
3 4 5
1 1
1 1
2 2
2 3
4 3
样例输出
90
算法讨论
数学问题,对于那些有限制的位置,只要把每个位置分别可能的情况加起来,然后全部乘起来即可,其余位置用快速幂来解决。没有限制的位置可以用等差数列的通项公式计算得出。
#include <cstdio>
#include <algorithm>
using namespace std;
#define maxn 10000005
#define maxk 100005
struct edge
{
long long a,b;
}a[maxk];
long long n,m,k;
long long ss=1,s=0,base,r,ans;
long long pow1(long long a,long long b)
{
r=1; base=a;
while (b!=0)
{
if (b % 2)
r=r*base % 1000000007;
base=base*base % 1000000007;
b/=2;
}
return 0;
}
long long cmp(edge aa, edge bb)
{
return (aa.a < bb.a) || (aa.a == bb.a && aa.b < bb.b);
}
int main()
{
scanf("%lld%lld%lld",&n,&m,&k);
long long t1=0,t2=0,p=0;
for (long long i=1;i<=k;i++)
scanf("%lld%lld",&a[i].a,&a[i].b);
sort(a+1,a+k+1,cmp);
for (long long i=1;i<=k;i++)
{
if (a[i].a!=a[i-1].a)
{
p++;
s=ss;
ss=(ss*(((1+n)*n/2)%1000000007-a[i].b+1000000007)) % 1000000007;
}
if (a[i].a==a[i-1].a && a[i].b!=a[i-1].b)
{
ss=(ss-(s*a[i].b)%1000000007 + 1000000007) % 1000000007;
}
}
pow1(((1+n)*n/2)%1000000007,m-p);
r%=1000000007;
ss%=1000000007;
ans=(ss*r) % 1000000007;
printf("%lld",ans);
return 0;
}
Pixiv ID:60537542