ZJOI[2010]基站选址 线段树优化DP

ZJOI[2010]基站选址

Part -1

这是一个很神仙的dp题，也可以算是线段树优化dp的一道入门题（神仙说的）。

Part 0

首先得打出暴力的写法。要注意dp的无后效性。（还有注意要仔细看题）

定义$dp[i][k]$为对于前$i$个村落,建$k$个通讯基站，且最后一个建在第$i$个村庄的最小花费。

那么就有$dp[i][k]=min(dp[j][k-1]+cost[j][i]+C[i])(j<i)$。$cost[j][i]$表示在村庄$j$和$i$建了通讯基站之后$j+1--->i-1$的村庄中没有被覆盖的村庄的补偿费用之和。那么统计答案可以在最后在$dp[i][K]$加上$i+1--->n$的没有被覆盖的村庄的补偿费用之和。

这里有一种比较方便的算最后答案的方法，就是在加一个id为n+1的村庄，距离1村庄的距离为∞（赋值赋一个较大的值即可）S，C，W都为0。最后的答案就是$dp[n+1][K+1]$。

part 1

接下来就是优化了。我们不难发现$cost[j][i]$是限制我们复杂度的关键。然后我们观察$cost[j][i]$算到的需要补偿的村庄有什么特征。然后就发现这些的村庄的S+D比$i$小，且这些村庄的D-S都比$j$大(这不废话）。于是我们可以把这些村庄按照S+D排序，在决策到i这个村庄时，S+D<D[i]村庄才会对决策有影响。如果我们的前一个要从$dp[j][k-1]$状态转移，那么我们我们就要算上这S+D<D[i]且D-S>D[j]的村庄的贡献，如果我们看这个村庄X的影响，如果前置状态为$dp[j][k-1]$且j不在X的范围内，那么就要加上这个村庄的补偿。不难发现这些j组成了一个区间。那么我们就可以用线段树来处理一个村庄影响的前置状态。（当然是以k为dp分层的标准的情况之下）,然后进行转移即可。

当然k为1的DP值要先预处理出来。

（真的看不懂可以看代码，再看不懂可以评论）。

AC代码：

#include<cstdio>
#include<algorithm>
#include<cstring> 
#define M 20005
using namespace std;
int dis[M],cost[M],S[M],W[M];
void check_min(int &x,int y){if(x>y)x=y;}
int n,K;
struct node{
	int W,stid,endis;
	bool operator <(const node &x)const{return endis<x.endis;}
}A[M];
int Get_L(int x){//预处理一个村庄S范围内的最左边的村庄 
	int L=1,R=x,res=x;
	while(L<=R){
		int mid=(L+R)>>1;
		if(dis[x]-dis[mid]<=S[x]){
			res=mid;
			R=mid-1;
		}else L=mid+1;
	}
	return res;
}
int dp[M];
struct Segment_tree{
	struct node{
		int L,R,mn,Lazy;
		void Add(int d){mn+=d;Lazy+=d;}
	}tree[M<<2];
	void Up(int p){tree[p].mn=min(tree[p<<1].mn,tree[p<<1|1].mn);}
	void Down(int p){
		int& res=tree[p].Lazy;
		if(!res)return;
		tree[p<<1].Add(res);
		tree[p<<1|1].Add(res);
		res=0;
	}
	void Build(int L,int R,int p){
		tree[p].L=L,tree[p].R=R,tree[p].mn=tree[p].Lazy=0;
		if(L==R){
			tree[p].mn=dp[L];
			return;
		}
		int mid=(L+R)>>1;
		Build(L,mid,p<<1);
		Build(mid+1,R,p<<1|1);
		Up(p);
	}
	void Updata(int L,int R,int d,int p){
		if(L<=tree[p].L&&tree[p].R<=R){
			tree[p].Add(d);
			return;
		}
		int mid=(tree[p].L+tree[p].R)>>1;
		Down(p);
		if(R<=mid)Updata(L,R,d,p<<1);
		else if(L>mid)Updata(L,R,d,p<<1|1);
		else Updata(L,mid,d,p<<1),Updata(mid+1,R,d,p<<1|1);
		Up(p);
	}
	int Query(int L,int R,int p){
		if(L<=tree[p].L&&tree[p].R<=R)return tree[p].mn;
		int mid=(tree[p].L+tree[p].R)>>1;
		Down(p);
		if(R<=mid)return Query(L,R,p<<1);
		else if(L>mid)return Query(L,R,p<<1|1);
		else return min(Query(L,mid,p<<1),Query(mid+1,R,p<<1|1));
	}
}ST;//区间更新，区间求最大值的线段树 
int main(){
	int ans=0;
	scanf("%d%d",&n,&K);
	for(int i=2;i<=n;i++)scanf("%d",&dis[i]);dis[n+1]=2e9+7;
	for(int i=1;i<=n;i++)scanf("%d",&cost[i]);
	for(int i=1;i<=n;i++)scanf("%d",&S[i]);
	for(int i=1;i<=n;i++)scanf("%d",&W[i]),ans+=W[i];
	for(int i=1;i<=n;i++){
		A[i].W=W[i];
		A[i].stid=Get_L(i);//i的S能覆盖的最左边的村庄 
		A[i].endis=dis[i]+S[i];//i的S能覆盖的最右边的位置 
	}
	sort(A+1,A+n+1);//按endis排序 
	int id=1,sum=0;
	for(int i=1;i<=n+1;i++){//预处理只建一个基站时的dp数组 
		while(id<=n&&A[id].endis<dis[i])sum+=A[id].W,id++;
		dp[i]=cost[i]+sum;
	}
	ST.Build(1,n+1,1);
	for(int j=2;j<=K+1;j++){
		id=1; 
		for(int i=j;i<=n+1;i++){//上一层建了j-1个基站，故最后一个基站最前也在j-1个村庄 
			while(id<=n&&A[id].endis<dis[i]){
				if(A[id].stid>j-1)ST.Updata(j-1,A[id].stid-1,A[id].W,1);//在线段树上更新 
				id++;
			}
			dp[i]=cost[i]+ST.Query(j-1,i-1,1);
		}
		if(j!=K+1)ST.Build(1,n+1,1);//把这一层放在线段树上 
		check_min(ans,dp[n+1]);
	}
	printf("%d\n",ans);
	return 0;
}