题解 微信步数
题目分析
如果将每一维度都合在一起考虑,复杂度貌似无论如何都需要带上一个 (prod_{i=1}^kw_i) ,类似 CSP2020T3 的思路,如果把每一维都分开考虑可能可以降低一点复杂度。
对于维度 (i) 的某一个位置 (p(1le ple w_i)) ,我们可以求出一个值 (mbox{time}_i(p)) 表示在仅考虑维度 (i) 时小 C 走出场地需要的最少步数,如果小 C 无论如何都无法走出场地那么 (mbox{time}_i(p)=+infty) ,求出 (mbox{time}) 这个函数之后,小 C 从 ((a_1,a_2,dots,a_k)) 走出场地需要的步数就是 (min_{i=1}^kmbox{time}_i(a_i)) 。
对于某一维度 (i) ,我们并不关心 (mbox{time}_i(1dots w_i)) 这个数列所有值之间的顺序,所以我们可以给 (mbox{time}_i(1dots w_i)) 先排个序。
我们需要求的东西是:
直接求出所有的 (mbox{time}_i(j)) ,然后从小到大枚举每一个 (mbox{time}_i(j)) 就可以很容易求出上面这个式子的值了,此时我们就成功地将 (prod_{i=1}^kw_i) 的复杂度降为了 (sum_{i=1}^kw_i) 了,按照实现不同复杂度此时可以达到 (mathcal O((sum_{i=1}^kw_i)log_2(sum_{i=1}^kw_i))) 或 (mathcal O(k(sum_{i=1}^kw_i))) 或 (mathcal O((sum_{i=1}^kw_i)log_2 k)) ,应该可以拿到 (80 m pts) 。
要拿更高的分数,需要进一步观察 (mbox{time}_i(1dots w_i)) 这个数列的性质,可以找到两个数列 (mbox{num}_i(1dots p_i)) 以及 (mbox{sum}_i(1dots s_i)) 使得 (mbox{time}_i(1dots w_i)) 是数列 (mbox{num}_i(1dots p_i)) 和数列 (mbox{sum}_i(1dots s_i)) 不断重复同时加上若干倍的 (n) 得到的:
并且上面这个序列是排好序的,有 (mbox{num}_i(j)le n,mbox{sum}_i(j)le n) 。
由于 (sum_{i=1}^kp_ile n) ,所以我们可以用上面的方法求出 (mbox{num}_i()) 数列这一段的值,即这个式子的值:
需要注意的是 (mbox{sum}_i()) 这个数列重复是一直重复到 (w_i) 为止的,如果把 (mbox{sum}_i(1)+tn,dots mbox{sum}_i(s_i)+tn) 称作是一轮的话,那么可能最后并不能凑成一整轮,由于 (sum_{i=1}^ks_ile n) ,所以无法凑够一整轮的这一部分也可以单独算,关键是算中间重复着的若干轮。
中间重复着的若干轮数列之间的 (mbox{sum}_i()) 的大小关系都是固定的,所以我们只需要模拟第一轮其实就已经模拟了所有轮了。
假设重复的轮数为 (r) ,对于当前枚举的某一个 (i) , (c_j=sum_{t=1}^{w_j}[mbox{time}_j(t)ge i]) ,那么此时造成贡献需要乘以的系数应该是这个:
令:
那么要求的就是:
如何求 (f(x)) ?自然数幂求和,设 (S_k(n)=sum_{i=0}^ni^k) :
直接 (mathcal O(k^2)) 求即可(当然可以用一些其它的方式求,这里不多说)。
如何求 (dp(x)) ?显然可以 (dp) 求,设 (dp(i,x)) 表示仅考虑前 (i) 个数的答案,那么转移就是考虑当前这个数是否包含在集合内:
最后的 (dp(x)=dp(k,x)) 。
如果每次都做一次 (dp) ,单次时间复杂度是 (mathcal O(k^2)) 的,总的时间复杂度就是 (mathcal O(nk^2)) 的,应该可以过,但是可以更优,不难发现这个 (dp) 是可撤销的,也就是我们可以删除一个数,即:
由于每次我们只会更改某一个位置 (v) ,即令 (c_vgets c_v-1) ,所以我们可以先撤销 (v) 造成的影响,然后更新完 (c_v) 后再加上 (v) 造成的影响,这样就不需要每次都 (dp) 一遍,单词时间复杂度降为 (mathcal O(k)) ,总时间复杂度为 (mathcal O(nk)) 。
如果算上所有操作,总时间复杂度就是 (mathcal O(k^2+n(k+log_2w))) 的。
总感觉自己把挺简单的一个东西讲得好复杂啊。
参考代码
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ch() getchar()
#define pc(x) putchar(x)
template<typename T>void read(T&x){
static char c;static int f;
for(c=ch(),f=1;c<'0'||c>'9';c=ch())if(c=='-')f=-f;
for(x=0;c>='0'&&c<='9';c=ch())x=x*10+(c&15);x*=f;
}
template<typename T>void write(T x){
static char q[64];int cnt=0;
if(x==0)return pc('0'),void();
if(x<0)pc('-'),x=-x;
while(x)q[cnt++]=x%10+'0',x/=10;
while(cnt--)pc(q[cnt]);
}
const int mod=1000000007,maxn=500006,maxk=12;
int mo(const int x){
return x>=mod?x-mod:x;
}
int power(int a,int x){
int re=1;
while(x){
if(x&1)re=1ll*re*a%mod;
a=1ll*a*a%mod,x>>=1;
}
return re;
}
int cnto[maxk][maxn],cntr[maxk][maxn];
int pos[maxk],w[maxk];
int ABS(int x){
return x<0?-x:x;
}
int num[maxk][maxn],p[maxk],pn[maxk];
int sum[maxk][maxn],s[maxk],cn[maxk];
int _s[maxk][maxk],f[maxk];
int dp[maxk];
void chkmin(int&x,int y){
x=min(x,y);
}
int main(){
freopen("walk.in","r",stdin);
freopen("walk.out","w",stdout);
int n,k;read(n),read(k);
for(int i=1;i<=k;++i)read(w[i]);
memset(cnto,0x3f,sizeof cnto);
memset(cntr,0x3f,sizeof cntr);
for(int i=1;i<=n;++i){
int c,d;read(c),read(d);pos[c]+=d;
if(pos[c]<0&& -pos[c]<=w[c])chkmin(cnto[c][-pos[c]],i);
if(pos[c]>0&&w[c]-pos[c]+1>=1 )chkmin(cntr[c][ pos[c]],i);
}
int ok=true,rd=0x3f3f3f3f;
for(int i=1;i<=k;++i){
int no=1,nr=1;
while(cnto[i][no]<0x3f3f3f3f)++no;--no;
while(cntr[i][nr]<0x3f3f3f3f)++nr;--nr;
s[i]=ABS(pos[i]);
if(no+nr>=w[i]){
int l=1,r=w[i];
int vl=min(cnto[i][l],cntr[i][w[i]-l+1]),
vr=min(cnto[i][r],cntr[i][w[i]-r+1]);
while(l<=r){
if(vl<vr)num[i][++p[i]]=vl,++l,vl=min(cnto[i][l],cntr[i][w[i]-l+1]);
else num[i][++p[i]]=vr,--r,vr=min(cnto[i][r],cntr[i][w[i]-r+1]);
}
for(int j=1;j<=s[i];++j)
sum[i][j]=0x3f3f3f3f;
ok=false;rd=0;
}
else{
int l=1,r=1;
while(l<=no||r<=nr){
if(r>nr||(l<=no&&cnto[i][l]<cntr[i][r]))
num[i][++p[i]]=cnto[i][l],++l;
else
num[i][++p[i]]=cntr[i][r],++r;
}
if(pos[i]>0){
for(int j=1;j<=s[i];++j)
sum[i][j]=cntr[i][nr-s[i]+j];
}
if(pos[i]<0){
for(int j=1;j<=s[i];++j)
sum[i][j]=cnto[i][no-s[i]+j];
}
}
if(s[i]>0){
ok=false;
rd=min(rd,(w[i]-p[i])/s[i]);
}
}
if(ok)return puts("-1"),0;
f[0]=rd;_s[0][0]=1;
for(int i=1;i<=k;++i){
_s[i][0]=0;
for(int j=1;j<=i;++j)
_s[i][j]=mo(_s[i-1][j-1]+mo(mod-1ll*(i-1)*_s[i-1][j]%mod));
f[i]=1;int tp=false;
for(int j=0;j<=i;++j){
if((rd-j)%(i+1)==0&&!tp)
f[i]=1ll*f[i]*((rd-j)/(i+1))%mod,tp=true;
else
f[i]=1ll*f[i]*(rd-j)%mod;
}
for(int j=0;j<i;++j)
f[i]=mo(f[i]+mo(mod-1ll*f[j]*_s[i][j]%mod));
}
for(int i=1;i<=k;i+=2)
f[i]=mo(mod-f[i]);
int ans=0;
int all=1;
for(int i=1;i<=k;++i)
all=1ll*all*w[i]%mod;
int pre=0;
while("Imakf ak ioi"){
int mn=0x3f3f3f3f,mp=0;
for(int i=1;i<=k;++i)
if(pn[i]<p[i]&&num[i][pn[i]+1]<mn)
mn=num[mp=i][pn[i]+1];
if(!all||mn>=0x3f3f3f3f){
ans=mo(ans+1ll*(n-pre)*all%mod);
pre=n;break;
}
ans=mo(ans+1ll*(mn-pre)*all%mod);
all=1ll*all*power(w[mp]-pn[mp],mod-2)%mod;
++pn[mp];all=1ll*all*(w[mp]-pn[mp])%mod;
pre=mn;
}
dp[0]=1;
for(int i=1;i<=k;++i){
cn[i]=w[i]-p[i];pn[i]=0;
for(int j=i;j>=1;--j)
dp[j]=mo(1ll*dp[j]*s[i]%mod+1ll*dp[j-1]*cn[i]%mod);
dp[0]=1ll*dp[0]*s[i]%mod;
}
pre=0;
while("Imakf ak ioi"){
int mn=0x3f3f3f3f,mp=0;
for(int i=1;i<=k;++i)
if(pn[i]<s[i]&&sum[i][pn[i]+1]<mn)
mn=sum[mp=i][pn[i]+1];
int add=0;
for(int i=0;i<=k;++i)
add=mo(add+1ll*dp[i]*f[k-i]%mod);
if(mn>=0x3f3f3f3f){
ans=mo(ans+1ll*(n-pre)*add%mod);
pre=n;break;
}
ans=mo(ans+1ll*(mn-pre)*add%mod);
++pn[mp];int I=power(cn[mp],mod-2);
for(int i=k;i>=1;--i)
dp[i]=1ll*mo(mod-1ll*dp[i+1]*s[mp]%mod+dp[i])*I%mod;
--cn[mp];
for(int i=1;i<=k;++i)
dp[i]=mo(1ll*dp[i]*cn[mp]%mod+1ll*dp[i+1]*s[mp]%mod);
pre=mn;
}
all=1;
for(int i=1;i<=k;++i){
w[i]-=(p[i]+rd*s[i]);
all=1ll*all*w[i]%mod;
if(s[i]&&w[i]<=s[i])
s[i]=w[i];
pn[i]=0;
}
pre=0;
while("Imakf ak ioi"){
int mn=0x3f3f3f3f,mp=0;
for(int i=1;i<=k;++i)
if(pn[i]<s[i]&&sum[i][pn[i]+1]<mn)
mn=sum[mp=i][pn[i]+1];
if(!all||mn>=0x3f3f3f3f)break;
ans=mo(ans+1ll*(mn-pre)*all%mod);//puts("ok");
all=1ll*all*power(w[mp]-pn[mp],mod-2)%mod;
++pn[mp];all=1ll*all*(w[mp]-pn[mp])%mod;
pre=mn;
}
write(ans),pc('
');
return 0;
}