【bzoj 2321】星器

傳送門~

解題思路

一道神題。
設（i，j）的勢能爲i^2+j^2，若將（i，j）和（i，k）進行2操作，可以發現始態和終態的勢能差剛好爲（j-k+1）*2。所以這題的正解就是，求出始態和終態的勢能差除以2就是答案。
代碼：

#include<algorithm>
#include<cmath>
#include<cstring>
#include<iostream>
#include<string>
#include<cstdio>
#include<cstdlib>
#define ll long long 
using namespace std;
ll n,m;
ll e;
int main(){
    ll x;
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++){
        scanf("%lld",&x);
        e+=x*(i*i+j*j);
    }
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++){
        scanf("%lld",&x);
        e-=x*(i*i+j*j);
    }
    printf("%lld",e/2);
    return 0;
}