描述
小 C 喜欢跑步,并且非常喜欢在微信步数排行榜上刷榜,为此他制定了一个刷微信步数的计划。
他来到了一处空旷的场地,处于该场地中的人可以用 k 维整数坐标 (a1,a2,…,ak) 来表示其位置。场地有大小限制,第 i 维的大小为 wi,因此处于场地中的人其坐标应满足 1≤ai≤wi(1≤i≤k)。
小 C 打算在接下来的 P=w1×w2×⋯×wk 天中,每天从场地中一个新的位置出发,开始他的刷步数计划(换句话说,他将会从场地中每个位置都出发一次进行计划)。
他的计划非常简单,每天按照事先规定好的路线行进,每天的路线由 n 步移动构成,每一步可以用 ci 与 di 表示:若他当前位于 (a1,a2,…,aci,…,ak),则这一步他将会走到 (a1,a2,…,aci+di,…,ak),其中 1≤ci≤k,di∈{−1,1}。小 C 将会不断重复这个路线,直到他走出了场地的范围才结束一天的计划。(即走完第 n 步后,若小 C 还在场内,他将回到第 1 步从头再走一遍)。
小 C 对自己的速度非常有自信,所以他并不在意具体耗费的时间,他只想知道 P 天之后,他一共刷出了多少步微信步数。请你帮他算一算。
输入描述
第一行两个用单个空格分隔的整数 n,k。分别表示路线步数与场地维数。
接下来一行 k 个用单个空格分隔的整数 wi,表示场地大小。
接下来 n 行每行两个用单个空格分隔的整数 ci,di,依次表示每一步的方向,具体意义见题目描述。
输出描述
仅一行一个整数表示答案。答案可能很大,你只需要输出其对 109+7 取模后的值。
若小 C 的计划会使得他在某一天在场地中永远走不出来,则输出一行一个整数 −1。
用例输入 1
3 2 3 3 1 1 2 -1 1 1
用例输出 1
21
用例输入 2
5 4 6 8 6 5 3 1 2 1 1 1 2 1 2 -1
用例输出 2
10265
提示
**【样例 #1 解释】**
从 (1,1) 出发将走 2 步,从 (1,2) 出发将走 4 步,从 (1,3) 出发将走 4 步。
从 (2,1) 出发将走 2 步,从 (2,2) 出发将走 3 步,从 (2,3) 出发将走 3 步。
从 (3,1) 出发将走 1 步,从 (3,2) 出发将走 1 步,从 (3,3) 出发将走 1 步。
共计 21 步。
**【数据范围】**
| 测试点编号 | n≤ | k≤ | wi≤ |
|:-:|:-:|:-:|:-:|
| 1∼3 | 5 | 5 | 3 |
| 4∼6 | 100 | 3 | 10 |
| 7∼8 | 105 | 1 | 105 |
| 9∼12 | 105 | 2 | 106 |
| 13∼16 | 5×105 | 10 | 106 |
| 17∼20 | 5×105 | 3 | 109 |
对于所有测试点,保证 1≤n≤5×105,1≤k≤10,1≤wi≤109,di∈{−1,1}。
上代码
#include <cstdio>
#include <iostream>
using namespace std;
const int M = 15;
const int N = 500005;
const int MOD = 1e9+7;
#define int long long
int read()
{
int x=0,f=1;char c;
while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
return x*f;
}
int n,k,inv[M],s[M][M],w[M],a[M],dt[M];
int ans,c[N],d[N],l[M],r[M],z[M];
int Abs(int x)
{
if(x>0) return x;
return -x;
}
int walk(int x,int y)//x维上走一步
{
z[x]+=y;
if(l[x]>z[x] || r[x]<z[x])
{
l[x]=min(l[x],z[x]);
r[x]=max(r[x],z[x]);
return 1;
}
return 0;
}
void init(int k)
{
s[0][0]=inv[0]=inv[1]=1;
for(int i=1;i<=k;i++)//第二类斯特林数
for(int j=1;j<=k;j++)
s[i][j]=(s[i-1][j-1]+s[i-1][j]*j)%MOD;
for(int i=2;i<=k+1;i++)
inv[i]=(MOD-MOD/i)*inv[MOD%i]%MOD;
}
int cal(int k,int n)//这里的n其实是开区间
{
int sum=0,C=1;
for(int i=0;i<=k;i++)//C(n,i+1)*i!*s[k][i]
{
C=C*max(0ll,n-i)%MOD;
sum=(sum+C*inv[i+1]%MOD*s[k][i])%MOD;
}
return sum;
}
int work()
{
int lim=MOD,res=0;
for(int i=1;i<=k;i++)
if(dt[i]) lim=min(lim,(a[i]+dt[i]-1)/dt[i]);
int dp[M]={1};
for(int i=1;i<=k;i++)//暴力多项式乘法
for(int j=i;j>=0;j--)
{
dp[j+1]=(dp[j+1]-dt[i]*dp[j])%MOD;
dp[j]=dp[j]*a[i]%MOD;
}
for(int i=0;i<=k;i++)
res=(res+dp[i]*cal(i,lim))%MOD;
return res;
}
signed main()
{
n=read();k=read();
init(k);
for(int i=1;i<=k;i++)
w[i]=read();
for(int i=1;i<=n;i++)
{
c[i]=read();d[i]=read();
if(walk(c[i],d[i]) && r[c[i]]-l[c[i]]<=w[c[i]])
//第二个判断很重要,因为要保证可能在这一步走出去
{
int x=1;
for(int j=1;j<=k;j++)
if(j!=c[i])
x=(x*max(0ll,w[j]-r[j]+l[j]))%MOD;
ans=(ans+i*x)%MOD;//要在这里走出去只有一种选择
}
}
int fl=1;
for(int i=1;i<=k;i++)
{
fl&=(z[i]==0);
dt[i]=Abs(z[i]);//别忘了
}
if(fl==1)
{
for(int i=1;i<=k;i++)
fl|=(r[i]-l[i]>=w[i]);
if(fl) puts("-1");
else printf("%lld\n",ans);
return 0;
}
for(int i=1;i<=k;i++)
a[i]=max(0ll,w[i]-r[i]+l[i]);
ans=(ans+n*work())%MOD;//第一部分贡献
for(int i=1;i<=n;i++)//第二部分贡献
if(walk(c[i],d[i]) && r[c[i]]-l[c[i]]<=w[c[i]])
{
fl=1;
for(int j=1;j<=k;j++)
if(j!=c[i])
fl&=(w[j]-r[j]+l[j]>0);
if(!fl) continue;
for(int j=0;j<k;j++)
if(j!=c[i]) a[j]=max(0ll,w[j]-r[j]+l[j]);
a[c[i]]=w[c[i]]-r[c[i]]+l[c[i]]+1;//差分差分
ans=(ans+i*work())%MOD;
a[c[i]]=w[c[i]]-r[c[i]]+l[c[i]];
ans=(ans-i*work())%MOD;
}
printf("%lld\n",(ans+MOD)%MOD)
}
//本文代码用的以前的