【BZOJ】2657: [Zjoi2012]旅游(journey)（树的直径）

题目

传送门：QWQ

分析

在任意两个不相邻的点连一条线，求这条线能穿过几个三角形。

建图比较讲究（详见代码）

求树的直径。

代码

#include <bits/stdc++.h>
using namespace std;
const int maxn=;
struct Node{
    int x,y,id;
}E[maxn];
int cmp(Node a,Node b){ return a.x<b.x||(a.x==b.x&&a.y<b.y); }
struct Edge{
    int u,v,dis;
};
vector<int> G[maxn];
int inq[maxn],d[maxn],n;
void Add(int u,int v){
    G[u].push_back(v); G[v].push_back(u);
}
queue<int> que;
int spfa(int s){
    memset(d,-,sizeof(d));
    memset(inq,,sizeof(inq));
    d[s]=;
    que.push(s);  inq[s]=;
    while(!que.empty()){
        int u=que.front(); que.pop();
        for(int i=;i<G[u].size();i++){
            int v=G[u][i];
            if(d[v]<d[u]+){
                d[v]=d[u]+;
                if(!inq[v]){ que.push(v); inq[v]=; }
            }
        }
    }
    int ans=;
    for(int i=;i<=n;i++) ans=max(ans,d[i]);
    return ans;
}
inline int in(){
    int x = , f = ; char ch = getchar();
    while (ch < '' || ch > ''){if (ch == '-')f = -;ch = getchar();}
    while (ch >= '' && ch <= ''){x = x *  + ch - '';ch = getchar();}
    return x * f;
}
int main(){
    scanf("%d",&n);
    int a,b,c,cnt=;
    for(int i=;i<n-;i++){
        a=in();b=in();c=in();
        if(a>b) swap(a,b); if(b>c) swap(b,c); if(a>b) swap(a,b);
        E[++cnt].x=a;E[cnt].y=b;E[cnt].id=i;
        E[++cnt].x=a;E[cnt].y=c;E[cnt].id=i;
        E[++cnt].x=b;E[cnt].y=c;E[cnt].id=i;
    }
    sort(E+,E++cnt,cmp);
    for(int i=;i<=cnt;i++){
        if(E[i].x==E[i-].x&&E[i].y==E[i-].y){
            Add(E[i].id,E[i-].id);
        }
    }
    int ans=,num;
    spfa();
    for(int i=;i<=n;i++){
        if(ans<d[i]){ ans=d[i]; num=i; }
    }
    printf("%d\n",spfa(num));
    return ;
}