【CF 1195E】OpenStreetMap

传送门
这道题就当做双重滑动窗口来做吧。首先,做出每一行长为b的滑动窗口,再用长为a的滑动窗口将先前的min存储,再做一次。
上马!

#include
#include
using namespace std;
typedef long long ll;
const int N = 3002;
int n, m, a, b;
ll G[N][N], x, y, z, g[N * N], r[N][N], ans;
deque <int> q;
void init() {
	for(int i = 1; i <= (n - 1) * m + m - 1; i ++) {
		g[i] = (g[i - 1] * x % z + y) % z;
	}
	for(int i = 1; i <= n; i ++) {
		for(int j = 1; j <= m; j ++) {
			G[i][j] = g[(i - 1) * m + j - 1];
		}
	}
}
void cal1() {
	for(int i = 1; i <= n; i ++) {
		while(! q.empty())
			q.pop_back();
		for(int j = 1; j <= m + 1; j ++) {//A
			if(j >= b + 1) {
				while(! q.empty() && q.front() < j - b)
					q.pop_front();
				r[i][j - b] = G[i][q.front()];//B
			}
			while(! q.empty() && G[i][q.back()] >= G[i][j])
				q.pop_back();
			q.push_back(j);//C
		}
	}
}
void cal2() {
	for(int j = 1; j <= m; j ++) {
		while(! q.empty())
			q.pop_back();
		for(int i = 1; i <= n + 1; i ++) {
			if(i >= a + 1) {
				while(! q.empty() && q.front() < i - a)
					q.pop_front();
				ans += r[q.front()][j];
			}
			while(! q.empty() && r[q.back()][j] >= r[i][j])
				q.pop_back();
			q.push_back(i);
		}
	}
}
int main() {
	scanf("%d %d %d %d %lld %lld %lld %lld", &n, &m, &a, &b, &g[0], &x, &y, &z);
	init();
	cal1();
	cal2();
	printf("%lld\n", ans);
	return 0;
}

注:
A:为了让后面的j-b最小为1,方便使用。
B:r数组表示第i行j为头长为b的滑动窗口的min。
C:虽然j号位可能并不是min,但因为窗口是滑动的,必须加入。
注:这道题卡空间!卡空间!卡空间!

如有问题,请大佬在评论区指出,谢谢!

你可能感兴趣的:(dp技巧)