Codeforces1295E Permutation Separation (线段树 + DP)

题目链接: Permutation Separation

大致题意

给定长度为$n$的序列$a$. $a$为1~n所有元素的一个排列.
给定长度为$n$的序列$b$, 表示操作第$i$个元素的花费为$b_i$.

你需要选定$k\in [1,n-1]$, 使得原序列分成两部分: l e f t = { a 1 , a 2 , . . . , a k } , r i g h t = { a k + 1 , a k + 2 , . . . , a n } left = \{ a_1, a_2, ..., a_k \}, right = \{ a_{k + 1}, a_{k + 2}, ..., a_n \} left={a1​,a2​,...,ak​},right={ak+1​,ak+2​,...,an​}
你可以执行的操作为: 把第$i$个元素移动到另外一个集合里去.
最终需要满足左集合中的最大元素值小于右集合中的最小元素值, 或某一集合为空集.

问: 最小的花费是多少.

解题思路

思维

首先考虑使某一集合为空集的情况: 只有当$left$ 或 $right$中只有一个元素时, 此时才有可能存在最优解.

否则, 我们考虑从左向右枚举$k$, 并讨论左集合$left$最大元素值$maxl$的情况:

既然从左向右枚举$k$, 那么最初的情况为: 所有元素都在$right$中.

设数组$c[]$, $c[i]$ 表示$maxl=i$时的最小花费.

当$maxl\in [a_k,n]$, 此时表明我们将$a_k$放在$left$中, 做了正贡献(和之前的所有情况相比). 那么此时的贡献应当是: $res1=min(c[maxl])-b_k$

当$maxl\in [1,a_k-1]$, 此时表明我们将$a_k$放在$left$中, 做了负贡献(和之前的所有情况相比). 那么此时的贡献应当是: $res2=min(c[maxl])+b_k$

因此当分割点为$k$时, 贡献为: $res=min(res1,res2)$.

---

线段树

我们发现上述信息本质是需要维护以权值为下标的区间最小值. 我们可以用权值线段树维护最小值.

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 2E5 + 10;
int a[N], b[N];
struct node {
	int l, r;
	ll val;
	ll lazy;
}t[N << 2];
void pushdown(node& op, ll lazy) { op.val += lazy, op.lazy += lazy; }
void pushdown(int x) {
	if (t[x].lazy == 0) return;
	pushdown(t[x << 1], t[x].lazy), pushdown(t[x << 1 | 1], t[x].lazy);
	t[x].lazy = 0;
}
void pushup(int x) { t[x].val = min(t[x << 1].val, t[x << 1 | 1].val); }

void build(int l, int r, int x = 1) {
	t[x] = { l, r, 0, 0 };
	if (l == r) return;
	int mid = l + r >> 1;
	build(l, mid, x << 1), build(mid + 1, r, x << 1 | 1);
}

void modify(int l, int r, ll c, int x = 1) {
	if (l <= t[x].l and r >= t[x].r) {
		pushdown(t[x], c);
		return;
	}
	pushdown(x);
	int mid = t[x].l + t[x].r >> 1;
	if (l <= mid) modify(l, r, c, x << 1);
	if (r > mid) modify(l, r, c, x << 1 | 1);
	pushup(x);
}

ll ask(int l, int r, int x = 1) {
	if (l <= t[x].l and r >= t[x].r) return t[x].val;
	pushdown(x);
	ll res = LLONG_MAX;
	int mid = t[x].l + t[x].r >> 1;
	if (l <= mid) res = ask(l, r, x << 1);
	if (r > mid) res = min(res, ask(l, r, x << 1 | 1));
	return res;
}
int main()
{
	int n; cin >> n;
	rep(i, n) scanf("%d", &a[i]);
	rep(i, n) scanf("%d", &b[i]);
	build(1, n);

	rep(i, n) modify(a[i], n, b[i]);

	ll res = min(b[1], b[n]);
	rep(i, n - 1) {
		if (a[i] != 1) modify(1, a[i] - 1, b[i]);
		modify(a[i], n, -b[i]);

		res = min(res, t[1].val);
	}

	cout << res << endl;

	return 0;
}

END