Codeforces(Vasya And The Mushrooms)

链接：http://codeforces.com/contest/1016/problem/C
思路：首先观察可知一共有n个终点，所以自然想到遍历一遍复杂度应该为O(n)，但是要求和，如果不加以处理必定会上升到O(n^2)的复杂度，考虑以下：sum123表示从左到右走到头所摘蘑菇总数，sun321表示从右到左所摘蘑菇总数，sum111表示这一段初始蘑菇总和。我们观察到，不管什么时候开始走sum123+sum321的情况(即一路走到头再从上面回来，中间不转弯)，都与原始求的sum123与sum321差一个常数*sum111，因为可把复杂度变为O(n)
代码：

#include
#include
using namespace std;

long long sum123[2][300001],sum321[2][300001],sum111[2][300001];
int n;
int a[2][300001];
int main(){
scanf("%d",&n);
for(int i=0;i<2;i++){
    for(int j=0;j=0;j--){
    sum123[i][j] = sum123[i][j+1] + j*1LL*a[i][j]; //从i,j一直走到右边端点所摘磨菇数总和
    sum321[i][j] = sum321[i][j+1] + (n-j)*1LL*a[i][j];//从右端点走到i,j所摘磨菇数总和
    sum111[i][j] = sum111[i][j+1] + a[i][j];//初始磨菇数总和
}
}
long long res = 0;
long long sum = 0;
for(int i=0,j=0;j