您的位置: 首页  >  IT文章  >  【BZOJ3672】【NOI2014】购票（线段树，斜率优化，动态规划） 【BZOJ3672】【NOI2014】购票（线段树，斜率优化，动态规划）

【BZOJ3672】【NOI2014】购票（线段树，斜率优化，动态规划） 【BZOJ3672】【NOI2014】购票（线段树，斜率优化，动态规划）

分类: IT文章 • 2022-04-15 14:17:44

题解

首先考虑(dp)的方程，设(f[i])表示(i)的最优值
很明显的转移(f[i]=min(f[j]+(dep[i]-dep[j])·p[i])+q[i])
其中满足(dep[i]-dep[j]le L[i])
然后就可以写出一个(O(n^2))的做法啦

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 222222
inline ll read()
{
    RG ll x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Line{int v,next,w;}e[MAX];
int h[MAX],cnt=1;
inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
int n,type,fa[MAX];
ll S[MAX],P[MAX],Q[MAX],L[MAX];
ll dep[MAX],f[MAX];
void dfs(int u)
{
	for(int i=h[u];i;i=e[i].next)
		dep[e[i].v]=dep[u]+e[i].w,dfs(e[i].v);
}
double Slope(int a,int b){return (f[a]-f[b])*1.0/(dep[a]-dep[b]);}
namespace Brute
{
	void DFS(int u)
	{
		if(u!=1)f[u]=1e18;
		for(int i=fa[u];i&&dep[u]-dep[i]<=L[u];i=fa[i])
			f[u]=min(f[u],f[i]+(dep[u]-dep[i])*P[u]+Q[u]);
		for(int i=h[u];i;i=e[i].next)DFS(e[i].v);
	}
	void Solve()
	{
		DFS(1);
		for(int i=2;i<=n;++i)printf("%lld
",f[i]);
	}
}
int main()
{
	n=read();type=read();
	for(int i=2;i<=n;++i)fa[i]=read(),S[i]=read(),P[i]=read(),Q[i]=read(),L[i]=read();
	for(int i=2;i<=n;++i)Add(fa[i],i,S[i]);
	dfs(1);
	if(n<=2000){Brute::Solve();return 0;}
}

如果没有下面的那条限制，我们可以很容易的写出一个效率优化
设(k<j<i)，(j)的转移优于(k)，那么有
(f[j]-dep[j]*p[i]<f[k]-dep[k]*p[i])
移项得到(p[i]>frac{f[j]-f[k]}{dep[j]-dep[k]})
很明显的斜率优化。
就这样我们就写出了一个(t=0)的(20)代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 222222
inline ll read()
{
    RG ll x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Line{int v,next,w;}e[MAX];
int h[MAX],cnt=1;
inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
int n,type,fa[MAX];
ll S[MAX],P[MAX],Q[MAX],L[MAX];
ll dep[MAX],f[MAX];
void dfs(int u)
{
	for(int i=h[u];i;i=e[i].next)
		dep[e[i].v]=dep[u]+e[i].w,dfs(e[i].v);
}
double Slope(int a,int b){return (f[a]-f[b])*1.0/(dep[a]-dep[b]);}
namespace Task0
{
	int St[MAX],top;
	int check(double K)
	{
		int l=2,r=top,ret=1;
		while(l<=r)
		{
			int mid=(l+r)>>1;
			if(Slope(St[mid],St[mid-1])>=K)r=mid-1;
			else l=mid+1,ret=mid;
		}
		return St[ret];
	}
	void Solve()
	{
		St[top=1]=1;
		for(int i=2;i<=n;++i)
		{
			int j=check(P[i]);
			f[i]=f[j]+Q[i]+(dep[i]-dep[j])*P[i];
			while(top>1&&Slope(i,St[top-1])<Slope(St[top],St[top-1]))--top;
			St[++top]=i;
			printf("%lld
",f[i]);
		}
	}
}
int main()
{
	n=read();type=read();
	for(int i=2;i<=n;++i)fa[i]=read(),S[i]=read(),P[i]=read(),Q[i]=read(),L[i]=read();
	for(int i=2;i<=n;++i)Add(fa[i],i,S[i]);
	dfs(1);
	if(type==0){Task0::Solve();return 0;}
}

当然，再把暴力给加上就有(40)分了。
（自己去拼接啊，要不然代码太多了）

剩下的部分分。
对于(t=1)，我们发现是没有距离限制的
显然是对于每一条链维护一个凸包，所以只需要维护一个可持久化栈然后在上面二分就好了。
这个东西似乎可以用主席树维护，然后在主席树上面二分就行了。
（我就懒得写了）
对于(t=2)，是一条链，但是有距离限制，
我们发现我们会从前面开始删去凸包上的一些点，这似乎非常不好做，
但是我们似乎可以用(CDQ)分治来做？按照能够用来更新的区间排序
每次(CDQ)的时候维护一段区间的凸包，然后更新一下答案就好了
（似乎可以这样做吧。。。）

现在终于可以来讲正解之一啦
现在的问题主要是如何动态维护凸包
其实我们把树直接树链剖分之后，对应的满足条件的是(dfs)序上的多段区间
所以我们用线段树暴力维护凸包，每个节点开一个(vector)
然后暴力维护这段区间的凸包
每个点最多会被(log)个线段树节点所包含，复杂度(O(nlogn))
然后每次询问的时候在跳重链+线段树+二分
似乎是三个(log)，但是我觉得只有两个(log)
似乎二分的(log)是加起来才有一个(log)吧。。。
代码如下：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 222222
#define lson (now<<1)
#define rson (now<<1|1)
inline ll read()
{
    RG ll x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Line{int v,next;ll w;}e[MAX];
int h[MAX],cnt=1;
inline void Add(int u,int v,ll w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
int n,type,fa[MAX];
ll S[MAX],P[MAX],Q[MAX],L[MAX];
ll dis[MAX],f[MAX];
int size[MAX],top[MAX],dfn[MAX],low[MAX],ln[MAX],tim,hson[MAX];
double Slope(int a,int b){return (f[a]-f[b])*1.0/(dis[a]-dis[b]);}
void dfs(int u)
{
	size[u]=1;
	for(int i=h[u];i;i=e[i].next)
	{
		int v=e[i].v;dis[v]=dis[u]+e[i].w;
		dfs(v);size[u]+=size[v];
		if(size[v]>size[hson[u]])hson[u]=v;
	}
}
void dfs(int u,int tp)
{
	top[u]=tp;dfn[u]=++tim;ln[tim]=u;
	if(hson[u])dfs(hson[u],tp);
	for(int i=h[u];i;i=e[i].next)
		if(e[i].v!=hson[u])dfs(e[i].v,e[i].v);
}
vector<int> t[MAX<<2];
void Modify(int now,int l,int r,int p)
{
	int tp=t[now].size();
	while(tp>1&&Slope(ln[p],t[now][tp-2])<Slope(t[now][tp-1],t[now][tp-2]))
		--tp,t[now].pop_back();
	t[now].push_back(ln[p]);
	if(l==r)return;
	int mid=(l+r)>>1;
	if(p<=mid)Modify(lson,l,mid,p);
	else Modify(rson,mid+1,r,p);
}
ll Calc(vector<int> t,int u)
{
	int l=1,r=t.size()-1,ret=0;
	while(l<=r)
	{
		int mid=(l+r)>>1;
		if(Slope(t[mid],t[mid-1])<1.0*P[u])l=mid+1,ret=mid;
		else r=mid-1;
	}
	int v=t[ret];
	return f[v]+(dis[u]-dis[v])*P[u]+Q[u];
}
ll Query(int now,int l,int r,int L,int R,int u)
{
	if(L<=l&&r<=R)return Calc(t[now],u);
	int mid=(l+r)>>1;ll ret=1e18;
	if(L<=mid)ret=min(ret,Query(lson,l,mid,L,R,u));
	if(R>mid)ret=min(ret,Query(rson,mid+1,r,L,R,u));
	return ret;
}
int Top=0;
void Jump(int u,int anc)
{
	f[u]=5e18;int U=u;u=fa[u];
	while(top[u]^top[anc])
	{
		f[U]=min(f[U],Query(1,1,n,dfn[top[u]],dfn[u],U));
		u=fa[top[u]];
	}
	f[U]=min(f[U],Query(1,1,n,dfn[anc],dfn[u],U));
}
void DFS(int u)
{
	S[++Top]=u;
	if(u!=1)
	{
		int l=1,r=Top-1,v=u;
		while(l<=r)
		{
			int mid=(l+r)>>1;
			if(dis[u]-dis[S[mid]]<=L[u])v=S[mid],r=mid-1;
			else l=mid+1;
		}
		Jump(u,v);
	}
	Modify(1,1,n,dfn[u]);
	for(int i=h[u];i;i=e[i].next)DFS(e[i].v);
	--Top;
}
int main()
{
	n=read();type=read();
	for(int i=2;i<=n;++i)fa[i]=read(),S[i]=read(),P[i]=read(),Q[i]=read(),L[i]=read();
	for(int i=2;i<=n;++i)Add(fa[i],i,S[i]);
	dfs(1);dfs(1,1);memset(S,0,sizeof(S));DFS(1);
	for(int i=2;i<=n;++i)printf("%lld
",f[i]);
	return 0;
}

相关推荐