3124: [Sdoi2013]直径

题目链接

题目大意：给定一棵树，求直径的长度以及有多少条边满足所有的直径都经过该边

题解：丢题解跑……

我的收获：分析性质……emmmmm

#include<cmath>
#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define FOR(a,b,c) for(int a=(b);a<=(c);a++)
using namespace std;

typedef long long LL;
const int N = 2*1e5+10;
struct Edge { int u,v,w;
};
vector<Edge> es;
vector<int> g[N];

int n,vis[N];

void read(int& x) {
    char c=getchar(); int f=1; x=0;
    while(!isdigit(c)) {if(c=='-')f=-1; c=getchar();}
    while(isdigit(c)) x=x*10+c-'0',c=getchar();
    x*=f;
}
void adde(int u,int v,int w) {
    es.push_back((Edge){u,v,w});
    g[u].push_back((int)es.size()-1);
}

LL maxdis; int maxu,pa[N];
void dfs(int u,int fa,LL d) {
    if(d>maxdis) maxdis=d,maxu=u;
    for(int i=0;i<g[u].size();i++) {
        Edge& e=es[g[u][i]];
        int v=e.v;
        if(v!=fa) {
            pa[v]=g[u][i];
            dfs(v,u,e.w+d);
        }
    }
}
void dfs2(int u,int fa,LL d) {
    if(d>maxdis) maxdis=d;
    for(int i=0;i<g[u].size();i++) {
        Edge& e=es[g[u][i]];
        int v=e.v;
        if(v!=fa && !vis[v]) dfs2(v,u,d+e.w); 
    }
}

int main() {
    read(n);
    int u,v,w;
    FOR(i,1,n-1) {
        read(u),read(v),read(w);
        adde(u,v,w),adde(v,u,w);
    }
    maxdis=0; dfs(1,-1,0); 
    int x=maxu;
    maxdis=0; dfs(x,-1,0); 
    LL ans=maxdis; int y=maxu;

    printf("%lld\n",ans);
    for(int i=y;i!=x;i=es[pa[i]].u) vis[i]=1;
    vis[x]=1;
    int l=0,r=0; LL nowdis=0;
    for(int i=y;i;i=es[pa[i]].u) {
        Edge& e=es[pa[i]];
        r++;
        maxdis=0; dfs2(i,-1,0);
        if(maxdis==nowdis) l=r;
        if(maxdis==ans-nowdis) break;
        nowdis+=e.w;
    }
    printf("%d",r-l);
    return 0;
}