☆P2491 [SDOI2011]消防-树的直径,LCA,dfs,倍增

某个国家有n个城市,这n个城市中任意两个都连通且有唯一一条路径,每条连通两个城市的道路的长度为zi(zi<=1000)。

这个国家的人对火焰有超越宇宙的热情,所以这个国家最兴旺的行业是消防业。由于政府对国民的热情忍无可忍(大量的消防经费开销)可是却又无可奈何(总统竞选的国民支持率),所以只能想尽方法提高消防能力。

现在这个国家的经费足以在一条边长度和不超过s的路径(两端都是城市)上建立消防枢纽,为了尽量提高枢纽的利用率,要求其他所有城市到这条路径的距离的最大值最小。

你受命监管这个项目,你当然需要知道应该把枢纽建立在什么位置上

https://www.luogu.org/problemnew/show/P2491

这道题先求出树的直径,然后我们从底部开始往上枚举,很显然的一个贪心:固定了一个端点之后,另一端点越远越好。所以我们直接枚举端点,找到它的另一端,左右指针扫直径。
这时候我通过LCAO(1)来计算距离,就可以把此情况的最大距离用O(n)处理得到。

注意这题的ans要设一个无穷大的值,0x3f3f3f能过,0x3f3f就过不了了......

#include
#include
#include
#include
using namespace std;
const int N=300001;
const int M=600001;
int n,s,cnt,maxn,root,t,l,r,sum,ans=0x3f3f3f;
struct node
{
    int next,to,v;
}e[M];
int head[N],dis[N],dep[N],f[N][21],lg[N];
bool vis[N];
inline void add(int x,int y,int w)
{
    e[++cnt].next=head[x];
    e[cnt].to=y;
    e[cnt].v=w;
    head[x]=cnt;
}
void dfs1(int u,int fa)
{
    for(int i=head[u];i;i=e[i].next){
        int to=e[i].to;
        if(to==fa) continue;
        dis[to]=dis[u]+e[i].v;
        dfs1(to,u);
    }
}
void dfs2(int u,int fa)
{
    for(int i=head[u];i;i=e[i].next){
        int to=e[i].to;
        if(to==fa) continue;
        dis[to]=dis[u]+e[i].v;
        f[to][0]=u;
        dep[to]=dep[u]+1;
        dfs2(to,u);
    }
}
int lca(int x,int y)
{
    if(dep[x]dep[y]){
		x=f[x][lg[dep[x]-dep[y]]-1];
	}
    if(x==y) return x;
    for(int i=lg[dep[x]];i>=0;i--)
    {
        if(f[x][i]!=f[y][i])
            x=f[x][i],y=f[y][i];
    }
    return f[x][0];
}
void init()
{
    for(int i=1;i<=9;i++){
        for(int j=1;j<=n;j++){
            f[j][i]=f[f[j][i-1]][i-1];
        }
    }
}
int main()
{
    scanf("%d%d",&n,&s);
    for(int i=1;i<=n;i++){
        lg[i]=lg[i-1]+(1<dis[root]) root=i;
    }
    memset(dis,0,sizeof(dis));
    dep[root]=1;
    dfs2(root,0);
    for(int i=1;i<=n;i++){
        if(!t||dis[i]>dis[t]) t=i;
    }
    init();
    l=t;r=t;vis[t]=1;
    while(l!=0){
        sum=0;
        if(dis[r]-dis[l]>s){
            vis[r]=0;
            r=f[r][0];
            vis[r]=1;	
        }
        else{
            while(dis[r]-dis[l]<=s&&l!=0){
                l=f[l][0];
                vis[l]=1;
            }
            int rlca;
            for(int i=1;i<=n;i++){
                if(vis[i]) continue;
                int lca1=lca(l,i),lca2=lca(r,i);
                if(dep[lca1]>dep[lca2]) rlca=lca1;
                else rlca=lca2;
                if(dep[rlca]

 

你可能感兴趣的:(精品题,noip历年省选,图论-LCA,搜索-bfs,倍增,图论-树的直径)