BZOJ3872: [Poi2014]Ant colony 树形DP

最新推荐文章于 2019-07-27 16:23:11 发布

Oakley_

最新推荐文章于 2019-07-27 16:23:11 发布

阅读量1k

点赞数 2

文章标签： BZOJ 树形DP

本文链接：https://blog.csdn.net/oakley_/article/details/52691719

版权

BZOJ 3872: [Poi2014]Ant colony

Time Limit: 30 Sec Memory Limit: 128 MB

Submit: 149 Solved: 56

[Submit][Status][Discuss]

题解： 难度适中的一个树形DP：我们可以发现从食蚁兽往四周搜是一个比较好的方法，于是我们就反着搜，mx[i]表示从这条边经过的蚂蚁最终想被食蚁兽吃掉最多能经过多少只，mn[i]就是最少能经过多少只，这样我们可以分别以第一条边的两个顶点作为树根TreeDP，最后得到所有叶子节点可以经过蚂蚁数量的范围。

将每群蚂蚁数量排序。然后对于每个叶子节点的限制范围，在所有蚂蚁群的蚂蚁数量上二分就可以。（时间限制很紧，TLE了两次无奈加上读入优化）

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define LL long long
const LL N=1000005;
const LL M=2000005;
LL n,m,k,a[N],leaf[N],tot,mx[N],mn[N];
int to[M],nxt[M],lj[N],du[N],cnt;
LL read()
{
    LL x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void insert(int f,int t)
{
	cnt++,to[cnt]=t,nxt[cnt]=lj[f],lj[f]=cnt;
	cnt++,to[cnt]=f,nxt[cnt]=lj[t],lj[t]=cnt;
}
int fa[N];
void dfs(int x)
{
	for(int i=lj[x];i;i=nxt[i])
	if(to[i]!=fa[x]) fa[to[i]]=x,du[x]++;
	for(int i=lj[x];i;i=nxt[i])
	if(to[i]!=fa[x])
	{
		mn[to[i]]=mn[x]*du[x];
		mx[to[i]]=(mx[x]+1)*du[x]-1;
		mx[to[i]]=min(mx[to[i]],a[m]);
		if(mn[to[i]]<=a[m]) dfs(to[i]);
	}
}
int cal(LL x)
{
	int l=1,r=m,ans=0;
	while(l<=r)
	{
		int mid=(l+r)>>1;
		if(a[mid]<x) ans=max(ans,mid),l=mid+1;
		else r=mid-1;
	}
	return ans;
}
int main()
{
	n=read(),m=read(),k=read();
	for(int i=1;i<=m;i++) a[i]=read();
	sort(a+1,a+m+1);
	LL s1,s2,x,y;
	s1=read(),s2=read();
	for(int i=2;i<n;i++)
	{
		x=read(),y=read();
		insert(x,y);
	}
	mn[s1]=mx[s1]=mn[s2]=mx[s2]=k;
	dfs(s1),dfs(s2);
	LL ans=0;
	for(int i=1;i<=n;i++)
	if(!du[i]) ans+=cal(mx[i]+1)-cal(mn[i]);
	printf("%lld",ans*k);
}