1. 程式人生 > >【BZOJ】2657: [Zjoi2012]旅遊(journey)(樹的直徑)

【BZOJ】2657: [Zjoi2012]旅遊(journey)(樹的直徑)

.com node 兩個 ID ont pac 傳送門 HR esp

題目

傳送門:QWQ

分析

在任意兩個不相鄰的點連一條線,求這條線能穿過幾個三角形。

建圖比較講究(詳見代碼)

求樹的直徑。

代碼

#include <bits/stdc++.h>
using namespace std;
const int maxn=200005;
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,-1,sizeof(d)); memset(inq,0,sizeof(inq)); d[s]=1; que.push(s); inq[s]=1; while(!que.empty()){
int u=que.front(); que.pop(); for(int i=0;i<G[u].size();i++){ int v=G[u][i]; if(d[v]<d[u]+1){ d[v]=d[u]+1; if(!inq[v]){ que.push(v); inq[v]=1; } } } } int ans=0; for(int i=1;i<=n;i++) ans=max(ans,d[i]);
return ans; } inline int in(){ int x = 0, f = 1; char ch = getchar(); while (ch < 0 || ch > 9){if (ch == -)f = -1;ch = getchar();} while (ch >= 0 && ch <= 9){x = x * 10 + ch - 0;ch = getchar();} return x * f; } int main(){ scanf("%d",&n); int a,b,c,cnt=0; for(int i=1;i<n-1;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+1,E+1+cnt,cmp); for(int i=1;i<=cnt;i++){ if(E[i].x==E[i-1].x&&E[i].y==E[i-1].y){ Add(E[i].id,E[i-1].id); } } int ans=0,num; spfa(1); for(int i=1;i<=n;i++){ if(ans<d[i]){ ans=d[i]; num=i; } } printf("%d\n",spfa(num)); return 0; }

【BZOJ】2657: [Zjoi2012]旅遊(journey)(樹的直徑)