牛客寒假6-F十字阵列

链接：https://ac.nowcoder.com/acm/problem/201986
来源：牛客网

题目描述

小 Q 新学会了一种魔法，可以对一个 N行M列 的网格上的敌人造成伤害

第 i 次使用魔法可以对网格上的一个十字形区域（即第 xi 行和第 yi 列的并）中的每个格子上的敌人造成 zi 点伤害

现在小 Q 一共使用了 H 次魔法，你需要在所有的施法完成之后统计造成伤害的情况，详见输出描述

提醒：本题输入规模较大，请使用高效的输入方式

1≤H≤500,000 1≤xi,yi,zi,N,M≤2000 1≤xi≤N,1≤yi≤M

输入描述:

第一行 3 个数字 N,M,H

接下来 H 行，每行 3 个正整数 xi，yi，zi

输出描述:

为了避免大量的输出，假设第 i 行第 j 列受到的总伤害是 w_ij

你只需要输出Σw_ij(i+j)对 10^9+7 取模的结果即可

示例1

输入

复制 5 5 5 1 1 1 2 2 2 3 3 3 4 4 4 5 5 5

5 5 5
1 1 1
2 2 2
3 3 3
4 4 4
5 5 5

输出

复制 890

890

说明

造成伤害的情况是：

1 3 4 5 6

3 2 5 6 7

4 5 3 7 8

5 6 7 4 9

6 7 8 9 5

补充的说明：

890 = 1*(1+1)+3*(1+2)+4*(1+3)+...+5*(5+5）,一共25项累加得到890

 

#include <bits/stdc++.h>
#define dbg(x) cout << #x << "=" << x << endl
#define eps 1e-8
#define pi acos(-1.0)

using namespace std;
typedef long long LL;

template<class T>inline void read(T &res)
{
    char c;T flag=1;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;res=c-'0';
    while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
}

namespace _buff {
    const size_t BUFF = 1 << 19;
    char ibuf[BUFF], *ib = ibuf, *ie = ibuf;
    char getc() {
        if (ib == ie) {
            ib = ibuf;
            ie = ibuf + fread(ibuf, 1, BUFF, stdin);
        }
        return ib == ie ? -1 : *ib++;
    }
}

int qread() {
    using namespace _buff;
    int ret = 0;
    bool pos = true;
    char c = getc();
    for (; (c < '0' || c > '9') && c != '-'; c = getc()) {
        assert(~c);
    }
    if (c == '-') {
        pos = false;
        c = getc();
    }
    for (; c >= '0' && c <= '9'; c = getc()) {
        ret = (ret << 3) + (ret << 1) + (c ^ 48);
    }
    return pos ? ret : -ret;
}

const int maxn = 2007;
const int M = 1000000007;

int n,m,h;

int a[maxn], b[maxn];
int ab[maxn][maxn];

int main()
{
    scanf("%d %d %d",&n, &m, &h);

    int ans = 0;
    int x, y, z;
    while(h--) {
        scanf("%d %d %d",&x,&y,&z);
        a[x] += z;
        b[y] += z;
        ab[x][y] += z;
    }
    for(int i = 1; i <= n; ++i) {
        for(int j = 1; j <= m; ++j) {
            //printf("val[%d][%d]:%lld
",i,j,a[i] + a[j] - ab[i][j]);
            ans = (ans + (LL)(a[i] + b[j] - ab[i][j]) * (i+j) % M) % M;
        }
    }
    printf("%d
",(ans+M)%M);
    return 0;
}