看这一题,第一眼就是前缀和优化,然后看了一眼提示,立马觉得前缀和优化还不够,他提示除以2,我就觉得跟这个有关系
然后想了半天不知道怎么优化,看了大佬的题解。瞬间顿悟了,原来是这样的:
假如数是a{1,2,3}
我们求1~3;
先看这个for(int i=l;i<=r;i++)
for(int j=l;j<=r;j++)
{
ans+=a[i][j];
}
1x1=1 1x2=2 1x3=3
2x1=2 2x2=4 2x3=6
3x1=3 3x2=6 3x3=9
我们要求的就是下面这一个
for(int i=1;i<=3;i++)
for(int j=i+1;j<=3;i++)
{
ans+=a[i][j];
}
2x1=2
3x1=3 3x2=6
就是上面正方形减去对角线除以2
然后我们看正方形的每一行,每一列的和都是一样的,用前缀和求出来即可。
//偷偷盗用一下大佬的代码@Bernard5
class Solution { public: static const int mod = 1000000007; typedef long long ll; vector<int> getSum(vector<int>& a, vector<int>& query) { ll size = a.size(), n = query.size(), x; ll sum[100007], tmp[100007]; sum[0] = tmp[0] = 0; vector<int> ans; for (int i = 0; i < size; ++i) { sum[i + 1] = (sum[i] + a[i]) % mod; tmp[i + 1] = (tmp[i] + 1LL * a[i] * a[i]) % mod; } for (int i = 0, l, r; i < n; i += 2) { l = query[i], r = query[i + 1]; x = (sum[r] - sum[l - 1]) % mod * (sum[r] - sum[l - 1]) % mod - (tmp[r] - tmp[l - 1]) % mod; x = x % mod * 500000004 % mod; if (x < 0) x += mod; ans.push_back(x); } return ans; } };
大佬门就是厉害,只有羡慕的份