50求助,找不出错误了
查看原帖
50求助,找不出错误了
501028
spider_oyster楼主2022/10/23 21:39
#include<bits/stdc++.h>
using namespace std;

const int p=998244353;
const int N=1e6+10;
int n,k,a[N],st[10000010];
long long power[N],sum[N];

void pre(int i)
{
	st[i]=st[i-1];
	while(st[i]>0&&a[st[i]]+i>-a[st[i]]) st[i]--;
}

inline long long calr(int i,int j,int m)
{
	return ((power[j]-power[i])%p+2ll*(sum[j]-sum[i])%p*m%p+1ll*m*m%p*(j-i)%p)%p;
}

int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		if(a[i]<0&&a[i]+i<=-a[i]) st[i]=i;
		else pre(i);
		power[i]=power[i-1]+1ll*a[i]*a[i];power[i]%=p;
		sum[i]=sum[i-1]+a[i];sum[i]%=p;
	}
	for(int i=n+1;i<=k;i++) pre(i);
	long long ans=0;
	for(int i=1;i<=k;i++) ans+=(calr(0,st[i],1)+calr(st[i],n,i))%p,ans%=p;
	printf("%lld",(ans%p+p)%p);
	return 0;
}
2022/10/23 21:39
加载中...