「THUPC 2021 初赛」区间矩阵乘法

link
$ans={\sum }_i{\sum }_j{\sum }_{k}a[p1+d\ast i+j]\times a[p2+d\ast j+k](0\le i,j,k<d)$

因为这个 $k$ 的加 $1$ 只会导致每次后面那个数的下标加 $1$，考虑前缀和。

令 $pre[i]=\sum a[k](k<=i)$

${\sum }_i{\sum }_{j}a[p1+d\ast i+j]\ast (pre[p2+d\ast j+d-1]-pre[p2+d\ast j-1])$

每次 $i$ 的变化只会影响一个值，老套路，移一下求和。

${\sum }_j\ast (pre[p2+d\ast j+d-1]-pre[p2+d\ast j-1])\ast {\sum }_{i}a[p1+d\ast i+j]$

再考虑前缀和，引入 $pre[i][j]$ 为公差为 $j$ 的前缀和。

${\sum }_j\ast (pre[p2+d\ast j+d-1][1]-pre[p2+d\ast j-1][1])\ast (pre[p1+d\ast (d-1)+j][d]-pre[p1+j-d][d])$（注意这里是减 $d$，不是减 $1$，因为公差是 $d$）

然后算一算时间复杂度：$\mathcal{O}(n\ln n+n\ast d)$，乍一看过不了，但题目保证 $pre$ 数组的下标不超过 $n$，即 $d^2\le n$ 则 $d\le \sqrt{n}$。

所以时间复杂度为：$\mathcal{O}(n\ln n+n\sqrt{n})$。有点卡常，需要把 $pre$ 数组小的一维放在前面，这样可以快一些。另外，对 $2^{32}$ 取模，开个 $unsignedint$ 就行了。

此题难点：1.想到用公差为 $j$ 的前缀和表示。2.发现 $d\le \sqrt{n}$。3.卡常。

Code

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <climits>
#define unsigned int
using namespace std;
const int MAXN = 2e5 + 5;
int n, m, a[MAXN], pre[450][MAXN], t, ans;
signed main() {
	int d, p1, p2;
	scanf("%u", &n); t = sqrt(n);
	for(int i = 1; i <= n; i ++) scanf("%u", &a[i]);
	for(int i = 1; i <= t; i ++) for(int j = 1; j <= n; j ++) pre[i][j] = (j - i >= 1 ? pre[i][j - i] : 0) + a[j];
	scanf("%u", &m);
	for(int i = 1; i <= m; i ++) {
		scanf("%u%u%u", &d, &p1, &p2); ans = 0;
		for(int j = 0; j < d; j ++) ans += (pre[1][p2 + d * j + d - 1] - pre[1][p2 + d * j - 1]) * (pre[d][p1 + d * (d - 1) + j] - (p1 + j - d >= 1 ? pre[d][p1 + j - d] : 0));
		printf("%u\n", ans);
	}
	return 0;
}