【IOI2018】会议【笛卡尔树】【dp】【线段树】

探讨了在长度为n的序列中,如何通过优化区间DP算法和利用笛卡尔树,实现对区间内钦定点与最大值间距离和的快速计算,解决了IOI级别的难题,复杂度达到O(nlogn)。

题意:长度为nnn的序列,qqq次询问,每次给定一个区间,钦定区间中的一个位置xxx,使得区间所有点 与xxx之间的最大值(含端点) 之和 最小,输出最小值。

n,q≤7.5×105n,q\leq7.5\times10^5n,q7.5×105

神仙题,不愧是IOI

首先有一个O(n2)O(n^2)O(n2)的 dp

f(l,r)=min⁡{f(l,k−1)+(r−k+1)hk,(k−l+1)hk+f(k+1,r)}f(l,r)=\min\{f(l,k-1)+(r-k+1)h_k,(k-l+1)h_k+f(k+1,r)\}f(l,r)=min{f(l,k1)+(rk+1)hk,(kl+1)hk+f(k+1,r)}

其中kkkl,rl,rl,r中的最大值的位置(如有多个随便选一个),即讨论钦定的点在最大值的左侧或右侧,然后另一侧的点贡献都是最大值

我觉得我考场上能想到这步就不错了

这个 dp 转移已经O(1)O(1)O(1)了,也不好压成一维,所以要么用可持久化之类的东西强行压状态,要么就不记录无用的状态

直接想的话两条路都不好走,但注意到这个过程实际上是最值分治,自然地想到笛卡尔树

哪里自然了啊kora

具体地讲,建出序列的笛卡尔树,然后在树上做上面dp,每个点只记录它代表的区间的dp值,这样就可以O(n)O(n)O(n)处理出来了。

然而就算处理出来了你仍然无法快速计算答案,因为询问区间可能会被拆成很多小段,你需要像平衡树一样沿着树递归下去。而笛卡尔树高是O(n)O(n)O(n)的,仍然可以被卡成狗。

不过思路感觉很对,考虑怎么优化

观察一下这个dp方程式

f(l,r)=min⁡{f(l,k−1)+(r−k+1)hk,f(k+1,r)+(k−l+1)hk}f(l,r)=\min\{f(l,k-1)+(r-k+1)h_k,f(k+1,r)+(k-l+1)h_k\}f(l,r)=min{f(l,k1)+(rk+1)hk,f(k+1,r)+(kl+1)hk}

套到树上:当前子树根结点是kkk,为了计算f(l,k−1)f(l,k-1)f(l,k1)f(k+1,r)f(k+1,r)f(k+1,r),我们需要继续往左右子树递归计算,我们这样子是不行的

但这个f(l,k−1)f(l,k-1)f(l,k1)f(k+1,r)f(k+1,r)f(k+1,r)比较特殊:它们都有一个端点是固定的!

为了叙述方便,下面只讨论f(k+1,r)f(k+1,r)f(k+1,r),左边的f(l,k−1)f(l,k-1)f(l,k1)是同理的

我们要是知道右子树的区间的所有前缀的dp信息就好了

看上去很扯,但实际上是可行的!

假设我们分别知道kkk的左右子树的前缀信息,也就是知道f(l,l...k−1)f(l,l...k-1)f(l,l...k1)f(k+1,k+1...r)f(k+1,k+1...r)f(k+1,k+1...r),现在考虑怎么合并f(l,l...r)f(l,l...r)f(l,l...r)

显然左边是不用管的

对于右边,我们再把这个方程式搬出来。为了看着顺眼,我把rrr换成了iii

f(l,i)=min⁡{f(l,k−1)+(i−k+1)hk,f(k+1,i)+(k−l+1)hk}f(l,i)=\min\{f(l,k-1)+(i-k+1)h_k,f(k+1,i)+(k-l+1)h_k\}f(l,i)=min{f(l,k1)+(ik+1)hk,f(k+1,i)+(kl+1)hk}

注意这个iii是在[k+1,r][k+1,r][k+1,r]内的,冷静分析一下,发现这个东西就是在原来的基础上整体加一个数,然后和一个一次函数取min⁡\minmin,因为是个区间,所以可以用线段树维护!

而这个方程是可以找到一个分界点,使得分界点左边取左边的值,右边取右边的值。原因是iii每增加111,左边的值固定增加hkh_khk,而右边增加f(k+1,i+1)−f(k+1,i)f(k+1,i+1)-f(k+1,i)f(k+1,i+1)f(k+1,i),区间往右扩张一位增加的代价总不可能大于区间最大值吧……所以左边迟早会超过右边,而且不会被反超。线段树上二分即可。

什么?线段树开不下?

但仔细想想会发现不同的位置是不会冲突的,即每个点xxx维护它当前处理的左端点到xxx的dp值,所以开一个线段树就可以了。

线段树合并应该也可以

f(l,k−1)f(l,k-1)f(l,k1)的话再开一棵线段树,左右倒过来就可以了

然后把询问离线,每组询问挂在区间最大值的点上面,建笛卡尔树的时候顺便处理一下就可以了。

复杂度O(nlog⁡n)O(n\log n)O(nlogn)

#include <iostream>
#include <cstring>
#include <cctype>
#include <cstdio>
#include <vector>
#define MAXN 750005
using namespace std;
inline int read()
{
	int ans=0;
	char c=getchar();
	while (!isdigit(c)) c=getchar();
	while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();
	return ans;
}
typedef long long ll;
int n;
struct SegmentTree
{
	#define lc p<<1
	#define rc p<<1|1
	struct node
	{
		int l,r,tag;
		ll k,b,lv,rv;
	}t[MAXN<<2];
	void build(int p,int l,int r)
	{
		t[p].l=l,t[p].r=r;
		if (l==r) return;
		int mid=(l+r)>>1;
		build(lc,l,mid),build(rc,mid+1,r);
	}
	inline void pushcov(int p,ll k,ll b)
	{
		t[p].k=k,t[p].b=b;
		t[p].lv=k*t[p].l+b,t[p].rv=k*t[p].r+b;
		t[p].tag=1;
	}
	inline void pushadd(int p,ll k,ll b)
	{
		t[p].k+=k,t[p].b+=b;
		t[p].lv+=k*t[p].l+b,t[p].rv+=k*t[p].r+b;
		if (!t[p].tag) t[p].tag=2;		
	}
	inline void pushdown(int p)
	{
		if (t[p].tag)
		{
			if (t[p].tag==1) pushcov(lc,t[p].k,t[p].b),pushcov(rc,t[p].k,t[p].b);
			if (t[p].tag==2) pushadd(lc,t[p].k,t[p].b),pushadd(rc,t[p].k,t[p].b);
			t[p].tag=t[p].k=t[p].b=0;
		}
	}
	inline void update(int p){t[p].lv=t[lc].lv,t[p].rv=t[rc].rv;} 
	ll query(int p,int k)
	{
		if (t[p].l==t[p].r) return t[p].lv;
		pushdown(p);
		if (k<=t[lc].r) return query(lc,k);
		return query(rc,k);
	}
	void search(int p,int l,int r,ll k1,ll b1,ll b2)
	{
		if (l<=t[p].l&&t[p].r<=r)
		{
			if (k1*t[p].l+b1<=t[p].lv+b2&&k1*t[p].r+b1<=t[p].rv+b2) return pushcov(p,k1,b1);
			if (t[p].lv+b2<=k1*t[p].l+b1&&t[p].rv+b2<=k1*t[p].r+b1) return pushadd(p,0,b2);
		}
		if (r<t[p].l||t[p].r<l) return;
		pushdown(p);
		search(lc,l,r,k1,b1,b2),search(rc,l,r,k1,b1,b2);
		update(p);
	}
}lt,rt;
int h[MAXN];
inline int Max(const int& x,const int& y){return h[x]>h[y]? x:y;}
int st[20][MAXN],LOG[MAXN];
inline int rmq(int l,int r)
{
	int t=LOG[r-l+1];
	return Max(st[t][l],st[t][r-(1<<t)+1]);
}
int ql[MAXN],qr[MAXN];
ll ans[MAXN];
vector<int> lis[MAXN];
void solve(int l,int r)
{
	if (l>r) return;
	int k=rmq(l,r);
	solve(l,k-1),solve(k+1,r);
	for (int i=0;i<(int)lis[k].size();i++)
	{
		int L=ql[lis[k][i]],R=qr[lis[k][i]];
		ans[lis[k][i]]=h[k]*(R-L+1ll);
		if (L<k) ans[lis[k][i]]=min(ans[lis[k][i]],rt.query(1,L)+(R-k+1ll)*h[k]);
		if (k<R) ans[lis[k][i]]=min(ans[lis[k][i]],(k-L+1ll)*h[k]+lt.query(1,R));
	}
	ll tl=rt.query(1,l),tr=lt.query(1,r);
	lt.search(1,k,r,h[k],tl+h[k]*(1ll-k),(k-l+1ll)*h[k]);
	rt.search(1,l,k,-h[k],tr+h[k]*(k+1ll),(r-k+1ll)*h[k]);
}
int main()
{
	n=read();
	int q=read();
	for (int i=1;i<=n;i++) h[i]=read();
	lt.build(1,1,n);rt.build(1,1,n);
	for (int i=1;i<=n;i++) st[0][i]=i;
	for (int j=1;j<20;j++)
		for (int i=1;i+(1<<(j-1))<=n;i++)
			st[j][i]=Max(st[j-1][i],st[j-1][i+(1<<(j-1))]);
	LOG[0]=-1;
	for (int i=1;i<=n;i++) LOG[i]=LOG[i>>1]+1;
	for (int i=1;i<=q;i++) ql[i]=read()+1,qr[i]=read()+1,lis[rmq(ql[i],qr[i])].push_back(i);
	solve(1,n);
	for (int i=1;i<=q;i++) printf("%lld\n",ans[i]);
	return 0;
}
# P5044 [IOI 2018] meetings 会议 ## 题目背景 本题为交互题,但在此请提交**完整程序**。 ## 题目描述 有 $N$ 座山横着排成一行,从左到右编号为从 $0$ 到 $N-1$。山的高度为 $H_i$($0\leq i\leq N-1$)。每座山的顶上恰好住着一个人。 你打算举行 $Q$ 个会议,编号为从 $0$ 到 $Q-1$。会议 $j$($0\leq j\leq Q-1$) 的参加者为住在从山 $L_j$ 到山 $R_j$(包括 $L_j$ 和 $R_j$)上的人($0\leq L_j\leq R_j\leq N-1$)。对于该会议,你必须选择某个山 $x$ 做为会议举办地($L_j\leq x\leq R_j$)。举办该会议的成本与你的选择有关,其计算方式如下: - 来自每座山 $y$($L_j\leq y\leq R_j$) 的参会者的成本,等于在山 $x$ 和 $y$ 之间(包含 $x$ 和 $y$)的所有山的最大高度。特别地,来自山 $x$ 的参会者的成本是 $H_x$,也就是山 $x$ 的高度。 - 会议的成本等于其所有参会者的成本之和。 你想要用最低的成本来举办每个会议。 注意,所有的参会者将在每次会议后回到他们自己的山;所以一个会议的成本不会受到先前会议的影响。 ## 输入格式 输入的第一行包含两个正整数 $N$ 和 $Q$,其意义见题目描述。 第二行包含 $N$ 个正整数 $H_0,H_1,\cdots, H_{N-1}$,表示这些山的高度。 第 $3+j$ 行($0\leq j\leq Q-1$),每行两个整数 $L_j, R_j$,表示这些会议的参会者的范围。 ## 输出格式 共 $Q$ 行,第 $1+j$ 行($0\leq j\leq Q-1$)一个整数 $C_j$,表示举办会议 $j $ 的最低的可能成本。 ## 输入输出样例 #1 ### 输入 #1 ``` 4 2 2 4 3 5 0 2 1 3 ``` ### 输出 #1 ``` 10 12 ``` ## 输入输出样例 #2 ### 输入 #2 ``` 3 3 2 1 2 0 0 0 1 0 2 ``` ### 输出 #2 ``` 2 3 5 ``` ## 输入输出样例 #3 ### 输入 #3 ``` 5 1 1000000000 1000000000 1 1000000000 1000000000 0 4 ``` ### 输出 #3 ``` 4000000001 ``` ## 输入输出样例 #4 ### 输入 #4 ``` 15 10 10 71 84 33 6 47 23 25 52 64 70 31 22 31 2 5 10 3 7 0 13 8 12 0 0 1 3 7 13 1 13 10 12 1 1 ``` ### 输出 #4 ``` 281 180 828 263 10 201 364 744 123 71 ``` ## 说明/提示 ### 样例#1解释 会议$j=0$有$L_j=0$和$R_j=2$,所以将由住在山$0$、$1$和$2$上的人参加。如果山$0$被选做举办地,会议$0$的成本计算如下: - 住在山$0$上的参会者的成本是$\max\lbrace H_0\rbrace=2$。 - 住在山$1$上的参会者的成本是$\max\lbrace H_0,H_1\rbrace=4$。 - 住在山$2$上的参会者的成本是$\max\lbrace H_0,H_1,H_2\rbrace=4$。 - 因此,会议$0$的成本是$2+4+4=10$。 不可能以更低的成本来举办会议$0$了,因此会议$0$的最低成本是$10$。 会议$j=1$有$L_j=1$和$R_j=3$,因此将由住在山$1$、$2$和$3$上的人参加。如果山$2$被选做举办地,会议$1$的成本计算如下: - 住在山$1$上的参会者的成本是$\max\lbrace H_1,H_2\rbrace=4$。 - 住在山$2$上的参会者的成本是$\max\lbrace H_2\rbrace=3$。 - 住在山$3$上的参会者的成本是$\max\lbrace H_1,H_2,H_3\rbrace=5$。 - 因此,会议$1$的成本是$4+3+5=12$。 不可能以更低的成本来举办会议$1$了,所以会议$1$的最低成本是$12$。 ### 限制条件 - $1\leq N\leq 750\space000$ - $1\leq Q\leq 750\space000$ - $1\leq H_i\leq1\space000\space000\space000$ - $0\leq L_j\leq R_j\leq R-1(0\leq j\leq Q-1)$ - $(L_j,R_j)\neq(L_k,R_k)(0\leq j<k\leq Q-1)$ ### 子任务 1. (4分) $N\leq3000,Q\leq10$ 2. (15分) $N\leq5000,Q\leq5000$ 3. (17分) $N\leq100\space000,Q\leq100\space000,H_i\leq2(0\leq i\leq N-1)$ 4. (24分) $N\leq100\space000,Q\leq100\space000,H_i\leq20(0\leq i\leq N-1)$ 5. (40分) 没有附加限制 ### Author Riku Kawasaki (Japan) ### Source IOI 2018 D2T3 为什么WA了(样例都过了) ```cpp #include <bits/stdc++.h> using namespace std; typedef long long ll; const int N = 750050; namespace SGT { ll left[N << 2], right[N << 2]; ll tagV[N << 2], tagK[N << 2], tagB[N << 2]; #define ls cur << 1 #define rs cur << 1 | 1 inline void add(int cur, ll v) { left[cur] += v, right[cur] += v; tagV[cur] += v; } inline void cover(int cur, int l, int r, ll k, ll b) { left[cur] = k * l + b, right[cur] = k * r + b; tagV[cur] = 0; tagK[cur] = k, tagB[cur] = b; } void pushdown(int cur, int l, int r) { if (~tagK[cur]) { int mid = l + r >> 1; cover(ls, l, mid, tagK[cur], tagB[cur]); cover(rs, mid + 1, r, tagK[cur], tagB[cur]); tagK[cur] = -1, tagB[cur] = 0; } if (tagV[cur]) { add(ls, tagV[cur]); add(rs, tagV[cur]); tagV[cur] = 0; } } void build(int cur, int l, int r) { left[cur] = right[cur] = 0; tagV[cur] = 0; tagK[cur] = -1, tagB[cur] = 0; if (l == r) return ; int mid = l + r >> 1; build(ls, l, mid); build(rs, mid + 1, r); } void update(int cur, int l, int r, ll k, ll b, ll v) { if (k * l + b >= left[cur] + v) { add(cur, v); return ; } if (k * r + b <= right[cur] + v) { cover(cur, l, r, k, b); return ; } pushdown(cur, l, r); int mid = l + r >> 1; update(ls, l, mid, k, b, v); update(rs, mid + 1, r, k, b, v); } void modify(int cur, int l, int r, int L, int R, ll k, ll b, ll v) { if (L <= l && r <= R) { update(cur, l, r, k, b, v); return ; } pushdown(cur, l, r); int mid = l + r >> 1; if (L <= mid) modify(ls, l, mid, L, R, k, b, v); if (mid + 1 <= R) modify(rs, mid + 1, r, L, R, k, b, v); } ll query(int cur, int l, int r, int idx) { if (l == r) return left[cur]; pushdown(cur, l, r); int mid = l + r >> 1; if (idx <= mid) return query(ls, l, mid, idx); else return query(rs, mid + 1, r, idx); } } int n, q, h[N], rk[N], L[N], R[N], p[N]; pair<int, int> a[N]; ll ansL[N], ansR[N]; vector<pair<int, int>> qry[N]; namespace ST { const int LG = 20; int st[N][LG]; inline int argmax(int x, int y) { return a[x] > a[y] ? x : y; } void build() { for (int i = 1; i <= n; i++) st[i][0] = i; for (int k = 1; k < LG; k++) for (int i = 1; i + (1 << k) - 1 <= n; i++) st[i][k] = argmax(st[i][k - 1], st[i + (1 << k - 1)][k - 1]); } inline int query(int l, int r) { int k = __lg(r - l + 1); return argmax(st[l][k], st[r - (1 << k) + 1][k]); } } int lc[N], rc[N], top, stk[N], root; void build_tree() { top = root = 0; for (int i = 1; i <= n; i++) { int k = top; while (k && a[i] > a[stk[k]]) k--; lc[i] = rc[i] = 0; if (k == 0) root = i; else rc[stk[k]] = i; if (k < top) lc[i] = stk[k + 1]; stk[top = ++k] = i; } } void solve(int u, int l, int r, ll ans[]) { if (!u) return ; solve(lc[u], l, u - 1, ans); solve(rc[u], u + 1, r, ans); ll tmp = (l < u ? SGT::query(1, 1, n, u - 1) : 0) - 1ll * (u - 1) * h[u]; SGT::modify(1, 1, n, u, r, h[u], tmp, 1ll * (u - l + 1) * h[u]); for (auto tmp : qry[u]) ans[tmp.second] = SGT::query(1, 1, n, tmp.first); } //fstream fans("C:/Users/admin/Downloads/P5044_2.out"); int main() { // freopen("C:/Users/admin/Downloads/P5044_2.in", "r", stdin); ios::sync_with_stdio(false); cin.tie(0), cout.tie(0); cin >> n >> q; for (int i = 1; i <= n; i++) { cin >> h[i]; a[i] = make_pair(h[i], i); } ST::build(); for (int i = 1; i <= q; i++) { cin >> L[i] >> R[i]; L[i]++, R[i]++; p[i] = ST::query(L[i], R[i]); } build_tree(); for (int i = 1; i <= q; i++) if (p[i] < R[i]) qry[rc[p[i]]].emplace_back(R[i], i); SGT::build(1, 1, n); solve(root, 1, n, ansR); reverse(h + 1, h + n + 1); for (int i = 1; i <= n; i++) a[i] = make_pair(h[i], -i); build_tree(); fill(qry + 1, qry + n + 1, vector<pair<int, int>>()); for (int i = 1; i <= q; i++) if (L[i] < p[i]) qry[rc[n - p[i] + 1]].emplace_back(n - L[i] + 1, i); SGT::build(1, 1, n); solve(root, 1, n, ansL); reverse(h + 1, h + n + 1); for (int i = 1; i <= q; i++) { ll sumL = ansL[i] + 1ll * (R[i] - p[i] + 1) * h[p[i]]; ll sumR = ansR[i] + 1ll * (p[i] - L[i] + 1) * h[p[i]]; cout << min(sumL, sumR) << '\n'; } return 0; } ```
最新发布
01-01
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值