牛客寒假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=;
     while((c=getchar())<''||c>'')if(c=='-')flag=-;res=c-'';
     while((c=getchar())>=''&&c<='')res=res*+c-'';res*=flag;
 }

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

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

 const int maxn = ;
 const int M = ;

 int n,m,h;

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

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

     int ans = ;
     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 = ; i <= n; ++i) {
         for(int j = ; j <= m; ++j) {
             //printf("val[%d][%d]:%lld\n",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\n",(ans+M)%M);
     return ;
 }