bzoj3246 [Ioi2013]Dreaming

Description

---

Serpent(水蛇)生活的地方有N个水坑，编号为0，…，N - 1，有M条双向小路连接这些水坑。每两个水坑之间至多有一条路径（路径包含一条或多条小路）相互连接，有些水坑之间根本无法互通(即M ≤ N-1 )。Serpent走过每条小路需要一个固定的天数，不同的小路需要的天数可能不同。Serpent的朋友袋鼠希望新修 N - M - 1条小路，让Serpent可以在任何两个水坑间游走。袋鼠可以在任意两个水坑之间修路，Serpent通过每条新路的时间都是L天。袋鼠希望找到一种修路方式使得修路之后Serpent在每两个水坑之间游走的最长时间最短。

n <= 500000

Solution

---

最近语文得学好一点，题意都看不懂了

观察题目给出的栗子可以发现，新添的边一定连接了不同的树（废话），并且每一条边连接的两点都是所在树上最长链的中点（绕，这是性质1
并且把单独的树看成是点后连接形成的新树是一个菊花图，这是性质2

知道了以上两个性质之后就能愉快地做题了。用树形dp求每个到其余任意点的最长路径，找到每棵树中最小的那个点连边就ok。由于我太弱了只想到了第一个性质而且写得丑只能瑟瑟发抖了

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) rep(i,0,n) x[i]=t;

const int N=500005;
const int E=1000505;

struct edge {int x,y,w,next;} e[E];

int ls[N],edCnt;
int f[N][2],rec[N];
int bel[N],tot;
int n,m,L,ans;

bool vis[N];

void add_edge(int x,int y,int w) {
    e[++edCnt]=(edge) {x,y,w,ls[x]}; ls[x]=edCnt;
    e[++edCnt]=(edge) {y,x,w,ls[y]}; ls[y]=edCnt;
    // printf("%d %d %d\n", x,y,w);
}

void dfs1(int now) {
    vis[now]=1;
    for (int i=ls[now];i;i=e[i].next) {
        if (vis[e[i].y]) continue;
        dfs1(e[i].y);
        int w=f[e[i].y][0]+e[i].w;
        if (w>f[now][0]) {
            rec[now]=e[i].y;
            f[now][1]=f[now][0];
            f[now][0]=w;
        } else if (w>f[now][1]) f[now][1]=w;
    }
}

void dfs2(int now,int root) {
    vis[now]=1;
    // if (f[now][0]<f[bel[tot]][0]||!bel[tot]) bel[tot]=now;
    if (f[now][0]<f[bel[root]][0]||!bel[root]) bel[root]=now;
    for (int i=ls[now];i;i=e[i].next) {
        if (vis[e[i].y]) continue;
        int w=f[now][0]+e[i].w;
        if (rec[now]==e[i].y) w=f[now][1]+e[i].w;
        if (w>f[e[i].y][0]) {
            rec[e[i].y]=now;
            f[e[i].y][1]=f[e[i].y][0];
            f[e[i].y][0]=w;
        } else if (w>f[e[i].y][1]) f[e[i].y][1]=w;
        dfs2(e[i].y,root);
    }
}

void dfs3(int now,int fa,int len,int &root) {
    if (len>ans) {
        ans=len; root=now;
    }
    for (int i=ls[now];i;i=e[i].next) {
        if (e[i].y==fa) continue;
        dfs3(e[i].y,now,len+e[i].w,root);
    }
}

int main(void) {
    freopen("data.in","r",stdin);
    freopen("std.out","w",stdout);
    scanf("%d%d%d",&n,&m,&L);
    rep(i,1,m) {
        int x,y,w; scanf("%d%d%d",&x,&y,&w);
        add_edge(++x,++y,w);
    }
    fill(vis,0); rep(i,1,n) if (!vis[i]) dfs1(i);
    // fill(vis,0); rep(i,1,n) if (!vis[i]) dfs2(i,++tot); 特别神奇，这样配套写会RE
    fill(vis,0); rep(i,1,n) if (!vis[i]) dfs2(i,i);
    // int mx=1;
    // rep(i,2,tot) if (f[bel[i]][0]>f[bel[mx]][0]) mx=i;
    // rep(i,1,tot) if (i!=mx) add_edge(bel[mx],bel[i],L);
    int mx=0;
    rep(i,1,n) if (bel[i]&&(f[bel[i]][0]>f[mx][0]||!mx)) mx=bel[i];
    rep(i,1,n) if (bel[i]&&bel[i]!=mx) add_edge(mx,bel[i],L);
    int tmp1,tmp2;
    ans=0; dfs3(1,0,0,tmp1);
    ans=0; dfs3(tmp1,0,0,tmp2);
    printf("%d\n", ans);
    return 0;
}