25pts
查看原帖
25pts
772909
Lovely_CCCyh___楼主2022/10/23 13:00
#include <bits/stdc++.h>
using namespace std;
int a[1000005];
int main()
{
	int n,k,sum=0;
	cin>>n>>k;
	for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=1;i<=k;i++)
	{
		int ans=0,j;
		for(j=1;j<=n;j++)
		{
			if(a[j]<0&&(a[j]+i)*(a[j]+i)<=(a[j]+1)*(a[j]+1)) ans=ans+(a[j]+1)*(a[j]+1);
			else break;
		}
		for(j=j;j<=n;j++) ans=ans+(a[j]+i)*(a[j]+i);
		sum=sum+ans;
		sum%=998244353;
	}
	cout<<sum;
	return 0;
}

2022/10/23 13:00
加载中...