【CF1016C】Vasya And The Mushrooms（模拟）

题意：给定一个2*n的矩阵，每一个点有一个权值，从左上角出发，时间t=0开始，连续的走，将矩阵走完，

每走一步，t++，并且得到t*当前格子的权值的值，求最大的权值和

n<=3e5，1<=a[i][j]<=1e6

思路：合法的构造方案如图

记录几个前缀和，真的不想写这种题

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
typedef long long ll;
using namespace std;
#define N   310000
#define oo  10000000
#define MOD 1000000007

ll s1[2][N],s2[2][N],s3[2][N],a[2][N];

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0;i<2;i++)
     for(int j=0;j<n;j++) scanf("%lld",&a[i][j]);
    for(int i=0;i<2;i++)
     for(int j=n-1;j>=0;j--)
     {
         s1[i][j]=s1[i][j+1]+a[i][j]*(j+1);
         s2[i][j]=s2[i][j+1]+a[i][j]*(n-j);
         s3[i][j]=s3[i][j+1]+a[i][j];
     }
    ll sum=0,ans=0;
    for(int i=0,j=0;j<n;j++,i^=1)
    {
        ll tmp=sum;
        tmp+=s1[i][j]+s3[i][j]*j;
        tmp+=s2[i^1][j]+s3[i^1][j]*(j+n);
        ans=max(ans,tmp);
        sum+=a[i][j]*(j*2+1);
        sum+=a[i^1][j]*(j*2+2);
    }
    for(int i=0;i<n;i++) ans-=a[0][i]+a[1][i];
    printf("%lld
",ans);
    return 0;
}