bzoj 1835: [ZJOI2010]base 基站选址（线段树优化dp）

1835: [ZJOI2010]base 基站选址

Time Limit: 100 Sec   Memory Limit: 64 MB
Submit: 1166   Solved: 548
[ Submit][ Status][ Discuss]

Description

有N个村庄坐落在一条直线上，第i(i>1)个村庄距离第1个村庄的距离为Di。需要在这些村庄中建立不超过K个通讯基站，在第i个村庄建立基站的费用为Ci。如果在距离第i个村庄不超过Si的范围内建立了一个通讯基站，那么就成它被覆盖了。如果第i个村庄没有被覆盖，则需要向他们补偿，费用为Wi。现在的问题是，选择基站的位置，使得总费用最小。 输入数据 (base.in) 输入文件的第一行包含两个整数N,K，含义如上所述。 第二行包含N-1个整数，分别表示D2,D3,…,DN ，这N-1个数是递增的。 第三行包含N个整数，表示C1,C2,…CN。 第四行包含N个整数，表示S1,S2,…,SN。 第五行包含N个整数，表示W1,W2,…,WN。

Input

输出文件中仅包含一个整数，表示最小的总费用。

Output

3 2 1 2 2 3 2 1 1 0 10 20 30

Sample Input

4

Sample Output

40%的数据中，N<=500；
100%的数据中，K<=N，K<=100，N<=20,000，Di<=1000000000，Ci<=10000，Si<=1000000000，Wi<=10000。

HINT

Source

Day1

[ Submit][ Status][ Discuss]

题解：线段树优化dp

先从O(N^2*K)的dp入手

f[i][j] 表示到第i个村庄建了j个基站的最小费用，i村庄必须建基站。

f[i][j]=min(f[k][j-1]+solve(k,i))+c[i]   其中solve(k+1,i-1) 表示中间村庄不建基站且未被覆盖的补偿。

对于这个dp,主要的瓶颈在于如何快速求解solve

因为这个dp 可以减去一维 ->  f[i]=min(f[k]+solve(k,i))+c[i]  只需要存储下上一层的状态就可以转移。

那么如果处理solve呢？

如果从solve(x,y)变成solve(x,y+1).
可以发现原来被左端点覆盖的村庄依然被覆盖,但是一些被右端点覆盖的村庄不会再被覆盖。所以我们可以用线段树维护f[x]+solve(x,y)的区间最小值，我们枚举y,然后logn的时间查询区间[1,y-1]的最小值，来更新当前状态的答案。

对于村庄i，我们利用lower_bound求出l[i],r[i]，表示在[l[i],r[i]]之间的村庄建基站，该村庄都会被覆盖。

每次处理完村庄x,我们枚举所有r[i]=x的车站i,然后在线段树中把[1,l[i]-1]全都加上w[i].
因为x不降,所以村庄i已经无法被右端点覆盖,那所有无法覆盖它的左端点的solve值都要加上w[i].

对于r[i]相同的车站，我们可以用用邻接表的方式来存储。

然后就可以枚举建立的基站数k,就上一层的f数组建立线段树，来更新当前的答案。

我们可以把n,k都+1，然后将d[n+1],s[n+1]赋值为inf,c[n+1]赋值为0，这样最优值就会存储在f[n+1]中，每次用f[n+1]更新最终答案即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 200003
#define inf 1000000003
using namespace std;
int n,m;
int s[N],c[N],d[N],l[N],r[N],tot,w[N],f[N];
int point[N],next[N],v[N],tr[N*4],delta[N*4];
void add(int x,int y)
{
	tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y;
}
void update(int x)
{
	tr[x]=min(tr[x<<1],tr[x<<1|1]);
}
void build(int now,int l,int r)
{
	delta[now]=0;
	if (l==r)
	{
	 tr[now]=f[l];
	 return;
    }
    int mid=(l+r)/2;
    build(now<<1,l,mid);
    build(now<<1|1,mid+1,r);
    update(now);
}
void pushdown(int now,int l,int r)
{
	if (!delta[now]) return ;
	tr[now<<1]+=delta[now];
	tr[now<<1|1]+=delta[now];
	delta[now<<1]+=delta[now];
	delta[now<<1|1]+=delta[now];
	delta[now]=0;
}
void pointchange(int now,int l,int r,int x,int v)
{
	if (l==r) 
	{
		tr[now]=v;
		return;
	}
	pushdown(now,l,r);
	int mid=(l+r)/2;
	if (x<=mid) pointchange(now<<1,l,mid,x,v);
	else pointchange(now<<1|1,mid+1,r,x,v);
	update(now);
}
void qjchange(int now,int l,int r,int ll,int rr,int v)
{
	if (l>=ll&&r<=rr)
	 {
	 	tr[now]+=v;
	 	delta[now]+=v;
	 	return;
	 }
	pushdown(now,l,r);
	int mid=(l+r)/2;
	if (ll<=mid) qjchange(now<<1,l,mid,ll,rr,v);
	if (rr>mid) qjchange(now<<1|1,mid+1,r,ll,rr,v);
	update(now);
}
int qjmin(int now,int l,int r,int ll,int rr)
{
	if (ll>rr)  return 0;
	if (l>=ll&&r<=rr)  return tr[now];
	pushdown(now,l,r);
	int mid=(l+r)/2;
	int ans=inf;
	if (ll<=mid)  ans=min(ans,qjmin(now<<1,l,mid,ll,rr));
	if (rr>mid) ans=min(ans,qjmin(now<<1|1,mid+1,r,ll,rr));
	return ans; 
}
int main()
{
	freopen("build.in","r",stdin);
	freopen("build.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=2;i<=n;i++)  scanf("%d",&d[i]);
	for (int i=1;i<=n;i++)  scanf("%d",&c[i]);
	for (int i=1;i<=n;i++)  scanf("%d",&s[i]);
	for (int i=1;i<=n;i++)  scanf("%d",&w[i]);
	d[n+1]=inf; w[n+1]=inf; s[n+1]=inf; n++; m++;
	for (int i=1;i<=n;i++)
	{
		l[i]=lower_bound(d+1,d+n+1,d[i]-s[i])-d;
		r[i]=lower_bound(d+1,d+n+1,d[i]+s[i])-d;
		if (d[r[i]]>d[i]+s[i])  r[i]--;
		add(r[i],i);
	}
	int ans=inf;
	for (int i=1;i<=m;i++)
	{
		if (i==1)
		{
			int t=0;
			for (int j=1;j<=n;j++)
			 {
			 	f[j]=t+c[j];
			 	for (int k=point[j];k;k=next[k])
			 	 t+=w[v[k]];
			 }
			ans=min(ans,f[n]);
			continue;
		}
		build(1,1,n);
		for (int j=1;j<=n;j++)
		{
			if (j!=1) f[j]=qjmin(1,1,n,1,j-1)+c[j];
			else f[j]=c[j];
		 	for (int k=point[j];k;k=next[k])
		 	 if (l[v[k]]-1>=1)
		 	  qjchange(1,1,n,1,l[v[k]]-1,w[v[k]]);
		}
		ans=min(ans,f[n]);
	}
	printf("%d\n",ans);
}