Light OJ 1369

题目链接：

http://lightoj.com/volume_showproblem.php?problem=1369

The problem you need to solve here is pretty simple. You are give a function f(A, n), where A is an array of integers and n is the number of elements in the array. f(A, n) is defined as follows:

long long f( int A[], int n ) { // n = size of A

    long long sum = 0;

    for( int i = 0; i < n; i++ )

        for( int j = i + 1; j < n; j++ )

            sum += A[i] - A[j];

    return sum;

}

Given the array A and an integer n, and some queries of the form:

1)      0 x v (0 ≤ x < n, 0 ≤ v ≤ 106), meaning that you have to change the value of A[x] to v.

2)      1, meaning that you have to find f as described above.

Input

Input starts with an integer T (≤ 5), denoting the number of test cases.

Each case starts with a line containing two integers: n and q (1 ≤ n, q ≤ 105). The next line contains n space separated integers between 0 and 106 denoting the array A as described above.

Each of the next q lines contains one query as described above.

Output

For each case, print the case number in a single line first. Then for each query-type "1" print one single line containing the value of f(A, n).

Sample Input	Output for Sample Input
1 3 5 1 2 3 1 0 0 3 1 0 2 1 1	Case 1: -4 0 4

Note

Dataset is huge, use faster I/O methods.

 

思路：把这N个数写到一个N*N的矩阵的对角线上，然后后面的减出来的数就写在后面，类似于下面的图，这样就可以发现，如果你改变某个数的值，例如咱们增加了他的值A，那么影响就是：这一行的每个数都+A，这一列的数都-A，这样就可以了，我们只要统计增加的个数就可以了，不要忘了更新数组

公式：

对于每个a[i]

sum+=(n-i-1)*a[i]-i*a[i];

n-i-1表示行，i表示列

This is the code：

#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<iomanip>
#include<list>
#include<map>
#include<queue>
#include<sstream>
#include<stack>
#include<string>
#include<set>
#include<vector>
using namespace std;
#define PI acos(-1.0)
#define EPS 1e-8
#define LL long long
#define ULL unsigned long long     //1844674407370955161
#define INT_INF 0x7f7f7f7f      //2139062143
#define LL_INF 0x7f7f7f7f7f7f7f7f //9187201950435737471
// ios::sync_with_stdio(false);
// 那么cin, 就不能跟C的 scanf，sscanf, getchar, fgets之类的一起使用了。
const int dr[]= {0, 0, -1, 1, -1, -1, 1, 1};
const int dc[]= {-1, 1, 0, 0, -1, 1, -1, 1};
int a[1000050];
int main()
{
    int T;
    scanf("%d",&T);
    for(int t=1; t<=T; ++t)
    {
        LL sum=0;
        LL n,k;
        scanf("%lld%lld",&n,&k);
        for(LL i=0; i<n; i++)
        {
            scanf("%lld",&a[i]);
            sum+=(n-i-1ll)*a[i]-i*a[i];//计算初始的和
            //n-i-1表示行，i表示列
        }
        printf("Case %d:\n",t);
        int tem;
        while(k--)
        {
            scanf("%d",&tem);
            if(!tem)
            {
                LL x,v;
                scanf("%lld%lld",&x,&v);
                LL y=v-a[x];
                sum+=((n-x-1ll)*y-x*y);
                a[x]=v;//不要忘了更新
            }
            else
                printf("%lld\n",sum);
        }
    }
    return 0;
}