bzoj3124: [Sdoi2013]直徑

Hanks_o發表於2018-04-22

題目傳送門

解法:
兩個傻逼結論:
首先肯定在同一條直徑上。因為每條直徑都要經過這一段。
肯定是連續的一段。
因為如果你有兩段的話。那麼敢問這兩段之間的路徑你是怎麼走過去的(樹上兩點路徑唯一!)

這樣的話就很好做了。
隨便找條直徑。
在這條直徑上確定一條路徑。使得這條路徑上:
任意一點都不可能從別的分支擴充套件出直徑。
懶得畫圖直接文字描述。
假設求出了一條直徑是1->2->5->7
那麼從2開始問(因為1肯定沒有兒子了,要不然它不會是直徑的端點)
那麼我們可以記錄1到2的距離,假設為t。
然後看看2能不能從別的分支走出一個距離t。如果可以那麼2之前的邊都不在所有直徑的交集上

問到5就記錄下1到5的距離,假設為T。
然後看看5能不能從別的分支走出一個T。可以的話那麼5之前的邊也都不行。
以此類推。

正著問一次倒著問一次。
就確定出了範圍。
然後就可以求解了呀。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
struct node {int x,y,next;ll d;}a[510000];int len,last[210000];
void ins(int x,int y,ll d) {len++;a[len].x=x;a[len].y=y;a[len].d=d;a[len].next=last[x];last[x]=len;}
int G,fa[210000],bian[210000];ll D;
void dfs(int x,ll d,int f) {
    if(D<d) {D=d;G=x;}
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=f){fa[y]=x;bian[y]=k;dfs(y,d+a[k].d,x);}
    }
}
int s[210000];
ll T;bool v[210000];
void dfs1(int x,ll d,int f) {
    if(T<d)T=d;
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(v[y]==true&&y!=f) dfs1(y,d+a[k].d,x);
    }
}
int ans;
void dfs2(int x,int d,int f,int ed) {
    if(x==ed){ans=d;return ;}
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=f) dfs2(y,d+1,x,ed);
    }
}
int main() {
    int n;scanf("%d",&n);len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++) {
        int x,y;ll d;scanf("%d%d%lld",&x,&y,&d);
        ins(x,y,d);ins(y,x,d);
    }
    int X,Y;
    memset(bian,0,sizeof(bian));dfs(1,0,0);X=G;D=0;dfs(X,0,0);Y=G;
    int x=Y,slen=0;memset(v,true,sizeof(v));
    while(x!=X) {s[++slen]=x;v[x]=false;x=fa[x];}
    s[++slen]=X;v[X]=false;
    int ansx=Y;ll K=a[bian[s[1]]].d;
    for(int i=2;i<=slen;i++) {
        T=0;dfs1(s[i],0,0);
        if(T==K)ansx=s[i];K+=a[bian[s[i]]].d;
    }
    for(int i=slen;i>=2;i--) bian[s[i]]=bian[s[i-1]];bian[s[1]]=0;
    int ansy=X;K=a[bian[s[slen]]].d;
    for(int i=slen-1;i>=1;i--) {
        T=0;dfs1(s[i],0,0);
        if(T==K)ansy=s[i];K+=a[bian[s[i]]].d;
    }dfs2(ansx,0,0,ansy);printf("%lld\n%d\n",D,ans);
    return 0;
}

相關文章