NOI.AC #111. 运气大战动态Dp 线段树维护矩阵乘法贪心

NOI.AC #111. 运气大战

题目传送门

分析

如果没有错排这个条件，根据排序不等式，肯定直接排序优秀。
有了错排这个条件，有一个神奇的结论，如果第 $i$ 个数匹配的是第 $j$ 个数，那么 $|i-j|\le 2$
排序之后考虑 $Dp$
如果 $i$ 和 $j$ 可以匹配，那么有
$f[i]=f[i-1]+w[i]\cdot r[i]$ (排序后)
注意到这样转移一定是最优的。
否则的话讨论一下错排，转移类似。
良心出题人没有卡常，可以过。
数据大一点的话，就要考虑动态 $Dp$
实际上转移方程式是一个带 $Max$ 的三阶递推。
用线段树维护矩阵乘法即可。
复杂度 $O(qn)$ 或 $O(qlogn)$

代码

暴力改

#include<bits/stdc++.h>
const int N = 3e4 + 10; const long long inf = 1e18;
int ri() {
	char c = getchar(); int x = 0, f = 1; for(;c < '0' || c > '9'; c = getchar()) if(c == '-') f = -1;
	for(;c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) - '0' + c; return x * f;
}
int n, q, w[N], r[N], c1[N], c2[N], loc[N];
long long cost[N][2], f[N];
bool cmp1(int u, int v) {return w[u] > w[v];}
bool cmp2(int u, int v) {return r[u] > r[v];}
void Mx(long long &a, long long b) {a = std::max(a,b);}
long long Calc(int a, int b) {
	if(a <= 0 || b <= 0 || a > n || b > n) return -inf;
	return 1LL * w[c1[a]] * r[c2[b]];
}
void Up(int x) {
	if(x <= 0 || x > n) return ;
	cost[x][0] = Calc(x - 1, x) + Calc(x, x - 1);
	cost[x][1] = std::max(Calc(x - 2, x - 1) + Calc(x - 1, x) + Calc(x, x - 2), 
		Calc(x, x - 1) + Calc(x - 1, x - 2) + Calc(x - 2, x));
}
int main() {
	n = ri(); q = ri();
	for(int i = 1;i <= n; ++i) w[i] = ri(), c1[i] = i;
	for(int i = 1;i <= n; ++i) r[i] = ri(), c2[i] = i;
	std::sort(c1 + 1, c1 + n + 1, cmp1);
	std::sort(c2 + 1, c2 + n + 1, cmp2);
	for(int i = 1;i <= n; ++i) loc[c2[i]] = i, Up(i);
	for(;q--;) {
		int u = ri(), v = ri();
		std::swap(r[u], r[v]); std::swap(loc[u], loc[v]);
		c2[loc[u]] = u; c2[loc[v]] = v;
		for(int i = 0;i < 3; ++i)
			Up(c2[u] + i), Up(c2[v] + i);
		for(int i = 1;i <= n; ++i) {
			f[i] = 0;
			if(i > 1) Mx(f[i], f[i - 2] + cost[i][0]);
			if(i > 2) Mx(f[i], f[i - 3] + cost[i][1]);
			if(c1[i] != c2[i])
				Mx(f[i], f[i - 1] + Calc(i, i));
		}
		printf("%lld\n", f[n]);
	}
	return 0;
}

线段树

#include<bits/stdc++.h>
const int N = 3e4 + 10, Nt = 32768;
int ri() {
	char c = getchar(); int x = 0, f = 1; for(;c < '0' || c > '9'; c = getchar()) if(c == '-') f = -1;
	for(;c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) - '0' + c; return x * f;
}
int n, q, w[N], r[N], loc[N];
struct D {int p, x;}a[N], b[N];
bool cmp(D a, D b) {return a.x < b.x;}
struct Maxtir {
	long long m[3][3];
	long long *operator [] (int x) {return m[x];}
	Maxtir() {memset(m, -0x3f, sizeof(m)); for(int i = 0;i < 3; ++i) m[i][i] = 0;}
	long long Calc(int x, int y) {return 1LL * a[x].x * b[y].x;}
	void Init(int x) {
		memset(m, -0x3f, sizeof(m));
		if(a[x].p != b[x].p) m[0][0] = Calc(x, x);
		if(x > 1) m[1][0] = Calc(x - 1, x) + Calc(x, x - 1);
		if(x > 2) m[2][0] = std::max(Calc(x - 2, x - 1) + Calc(x - 1, x) + Calc(x, x - 2), 
								Calc(x, x - 1) + Calc(x - 1, x - 2) + Calc(x - 2, x));
		m[0][1] = m[1][2] = 0;
	}
	Maxtir operator * (Maxtir a) {
		Maxtir c; memset(c.m, -0x3f, sizeof(c.m));
		for(int i = 0;i < 3; ++i)
			for(int j = 0;j < 3; ++j)
				for(int k = 0;k < 3; ++k)
					c[i][j] = std::max(c[i][j], m[i][k] + a[k][j]);
		return c;
	}
}T[Nt << 1];
void Up(int i) {
	if(i > n) return ;T[i + Nt].Init(i);
	for(i += Nt;i >>= 1;) T[i] = T[i << 1] * T[i << 1 | 1];
}
int main() {
	n = ri(); q = ri();
	for(int i = 1;i <= n; ++i) a[i] = (D){i, ri()};
	for(int i = 1;i <= n; ++i) b[i] = (D){i, ri()};
	std::sort(a + 1, a + n + 1, cmp);
	std::sort(b + 1, b + n + 1, cmp);
	for(int i = 1;i <= n; ++i) loc[b[i].p] = i;
	for(int i = 1;i <= n; ++i) T[i + Nt].Init(i);
	for(int i = Nt - 1; i; --i) T[i] = T[i << 1] * T[i << 1 | 1];
	for(;q--;) {
		int u = ri(), v = ri();
		std::swap(b[loc[u]].p, b[loc[v]].p);
		std::swap(loc[u], loc[v]);
		for(int i = 0;i < 3; ++i)
			Up(loc[u] + i), Up(loc[v] + i);
		printf("%lld\n", T[1][0][0]);
	}
	return 0;
}

NOI.AC #111. 运气大战 动态Dp 线段树维护矩阵乘法 贪心

NOI.AC #111. 运气大战

题目传送门

分析

代码

猜你喜欢

NOI.AC #111. 运气大战动态Dp 线段树维护矩阵乘法贪心