事后想出的方法,可以用单调栈。不过这个单调栈的构造是真的难恶心到我了,尤其出现0的那个情况一直过不去。最后才搞明白。
可以这么想:统计后面的人有多少步延迟了,刚开始想到这个地方就直接带到后缀和的思路去了,在不怎么正确的思路上越走越远。后来发现数据范围和动态性问题无法解决便放弃了它。再后来发现了一丢丢单调栈的特征(每次前k人移动的时候就重置了这K个人的步数,那么只用保存它之前的且大于K重置的记录就行了 /(ㄒoㄒ)/~~)。还有就是计数的时候,统计第i时,第i个人要单独拎出来算,因为他到柜台前就不用走了。
详细解释待题放出来再说。
代码如下(两个样例已过,至少不会超时。啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊)
如果有人看见题放出来了,麻烦帮忙交一发,评论区告诉我结果,拜托了
#include <stdio.h>
#include <algorithm>
#include <string>
#include <string.h>
#include <iostream>
#include <map>
using namespace std;
const int max_n=1e6+10;
typedef long long ll;
ll INF=0x3f3f3f3f;
ll sta[max_n];
ll ind[max_n];
ll res[max_n];
int main()
{
ll N,Q,K,sum=0;
int top=0;
scanf("%lld%lld",&N,&Q);
sta[top]=INF;
ind[top++]=0;
for(int i=1;i<=Q;i++)
{
ll sum=0;
scanf("%lld",&K);
ll j=i;
while(top&&sta[top-1]<=K+i)
{
if(sta[top-1]>=i)
{
if(j==i)
{sum+=(j-ind[top-1]-1); j++;}
sum+=(sta[top-1]-j+1)*((ll)i-ind[top-1]);
j=sta[top-1]+1;
}
top--;
}
if(j==i) {sum+=(ll)i-ind[top-1]-1; j+=1;}
if(j<=K+i) sum+=(K+i-j+1)*((ll)i-ind[top-1]);
sta[top]=K+i;
ind[top++]=i;
res[i]=sum;
}
for(int i=1;i<=Q;i++)
printf("%lld%c",res[i],i==Q?'\n':' ');
}