码迷,mamicode.com
首页 > 其他好文 > 详细

[ZJOI2010]基站选址

时间:2019-01-14 21:27:20      阅读:217      评论:0      收藏:0      [点我收藏+]

标签:辅助   $2   时间   问题   upd   signed   +=   str   cout   

 真毒瘤

这个题目耗了我半天。。结果是线段树打错了。。。


回归正题:线段树$+dp$

首先当然是先考虑朴素$dp$啦,相信你既然都来做这题了,朴素的方程自然不用我多说,设$f[i][j]$表示在前$i$个村庄内,第$j$个基站建在$i$处的最小费用(不考虑$i$~$n$的赔偿费用等)

方程为:
$$f[i][j]=min(f[k][j-1]+pay[k][i])$$
其中$pay[k][i]$表示从第$k$个村庄到第$i$个村庄的赔偿费用之和

那么我们观察上面这个方程,这个方程是$O(n^2k)$的,可以发现类似于背包问题,是可以滚掉一维,因为发现$f$数组第二维只跟上次的值也就是$j-1$次的值有关,那么我们只要直接利用上一次求出来的值继续$dp$就行了,不需要多开一维,所以$dp$方程自然就化为$$f[i]=min(f[k]+pay[k][i])$$

那么我们继续思考:主要的时间消耗在了哪里?自然是怎么快速计算$pay[k][i]$了

那么我们思考:

对于每一个村庄,都有一个范围内需要建立基站,否则就要赔偿,那么我们设第$i$个村庄的范围为$[L,R]$,如果正在考虑$R$处建不建基站,那么有下列情况:

$1$、不在$R$处设立基站,那么对于村庄$i$来说,上一个基站在$[1,L-1]$这个区间的话,就要赔偿村庄$i$了,因为$[L,R]$这个区间没有建基站,那么我们就要快速的在$[1,L-1]$中区间加村庄$i$的赔偿费用了,我们就以线段树为例啦

$2$、在$R$处建立基站,那么也就相当于最后一个基站设立在$[1,R-1]$这个区间中,找一个费用最小值来转移嘛,还是线段树$qwq$

所以我们要开一个线段树来维护$f+pay$的最小值,要有区间加法和区间查询的操作

那么为了上面的操作,我们还要开几个数组辅助(如果你是$dalao$当我没说

$st[i]$表示第$i$个村庄对应区间的左端点$L$

$ed[i]$表示第$i$个村庄对应区间的右端点$R$

那么当$ed_x=i$的时候,如果$i$处不建,那么就要区间加上$pay[x]$的值,然而可能有很多点的$ed$都是$i$,所以我们用链式前向星来保存,具体可以看代码实现

接下来就是美滋滋的代码时间~~~

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 20007
#define int long long
#define inf 0x3f3f3f3f3f3f3f3f
using namespace std;
struct Edge
{
	int to,nxt;
}edge[N<<2];
struct Tree
{
	int date,mark;
}tr[N<<2];
int n,k,cnt;
int dis[N],val[N],range[N],pay[N],st[N],ed[N],head[N],f[N];
void Add(int u,int v)
{
	edge[++cnt]=(Edge){v,head[u]};
	head[u]=cnt;
}
void Get()
{
	for(int i=1;i<=n;++i)
	{
		st[i]=lower_bound(dis+1,dis+1+n,dis[i]-range[i])-dis;
		ed[i]=lower_bound(dis+1,dis+1+n,dis[i]+range[i])-dis;
		if(dis[ed[i]]>dis[i]+range[i])
			--ed[i];
		Add(ed[i],i);
	}
}
void Pushup(int rt)
{
	tr[rt].date=min(tr[rt<<1].date,tr[rt<<1|1].date);
}
void Build(int rt,int l,int r)
{
	tr[rt].mark=0;
	if(l==r)
	{
		tr[rt].date=f[l];
		return;
	}
	int mid=l+((r-l)>>1);
	Build(rt<<1,l,mid);
	Build(rt<<1|1,mid+1,r);
	Pushup(rt);
}
void Pushdown(int rt)
{
	if(tr[rt].mark)
	{
		tr[rt<<1].date+=tr[rt].mark;
		tr[rt<<1|1].date+=tr[rt].mark;
		tr[rt<<1].mark+=tr[rt].mark;
		tr[rt<<1|1].mark+=tr[rt].mark;
		tr[rt].mark=0;
	}
}
int Search(int rt,int l,int r,int L,int R)
{
//	cout<<"l:"<<l<<" r:"<<r<<" L:"<<L<<" R:"<<R<<endl;
	if(L>R)
		return inf;
	if(L<=l&&r<=R)
		return tr[rt].date;
	int mid=l+((r-l)>>1);
	Pushdown(rt);
	int num=inf;
	if(L<=mid)
		num=min(num,Search(rt<<1,l,mid,L,R));
	if(mid<R)
		num=min(num,Search(rt<<1|1,mid+1,r,L,R));
	return num;
}
void Update(int rt,int l,int r,int L,int R,int c)
{
//	cout<<"l:"<<l<<" r:"<<r<<endl;
	if(L>R)
		return;
	if(L<=l&&r<=R)
	{
		tr[rt].date+=c;
		tr[rt].mark+=c;
		return;
	}
	Pushdown(rt);
	int mid=l+((r-l)>>1);
	if(L<=mid)
		Update(rt<<1,l,mid,L,R,c);
	if(mid<R)
		Update(rt<<1|1,mid+1,r,L,R,c);
	Pushup(rt);
}
void Dp()
{
	int now=0;
	for(int j=1;j<=n;++j)
	{
		f[j]=now+val[j];
		for(int p=head[j];p;p=edge[p].nxt)
		{
			int v=edge[p].to;
			now+=pay[v];
		}
	}
	int ans=f[n];
//	for(int i=1;i<=n;++i)
//	{
//		printf("%d ",f[i]);
//	}
	for(int i=2;i<=k;++i)
	{
		Build(1,1,n);
		for(int j=1;j<=n;++j)
		{
			f[j]=Search(1,1,n,1,j-1)+val[j];
		//	cout<<f[j]<<" ";
			for(int p=head[j];p;p=edge[p].nxt)
			{
				int v=edge[p].to;
				Update(1,1,n,1,st[v]-1,pay[v]);
			}
		}
		ans=min(ans,f[n]);//cout<<"ans:"<<ans<<endl;
	}
	printf("%lld",ans);
}
void Init()
{
	scanf("%lld%lld",&n,&k);
	for(int i=2;i<=n;++i)
		scanf("%lld",&dis[i]);
	for(int i=1;i<=n;++i)
		scanf("%lld",&val[i]);
	for(int i=1;i<=n;++i)
		scanf("%lld",&range[i]);
	for(int i=1;i<=n;++i)
		scanf("%lld",&pay[i]);
	++n;++k;
	dis[n]=pay[n]=inf;
}
signed main()
{
	Init();
	Get();
	Dp();
	return 0;
}

  

[ZJOI2010]基站选址

标签:辅助   $2   时间   问题   upd   signed   +=   str   cout   

原文地址:https://www.cnblogs.com/yexinqwq/p/10268709.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!