題目鏈接
題意，求一棵樹被所有直徑經(jīng)過的邊的條數(shù)。
這題是我們8.25KS圖論的最后一題，當時我果斷打了暴力求所有直徑然后樹上差分統(tǒng)計的方法，好像有點小問題，boom0了。
考完改這題，改了好久，各種各樣的小bug，至少有七八個。。。
思路：先隨便找一條直徑，然后從一個端點開始遍歷這條直徑，如果當前點能分叉出一條直徑，那么這條直徑后面的點都不可能被所有直徑穿過了，于是我們找到第一個能分叉的點，然后再從這個點往回遍歷，找到第一個能分叉的點，這2個點中間的路徑的邊即為所求。
代碼就算了吧，打這題時我就沒想過要有可讀性

#include <cstdio> #include <algorithm> #include <queue> using namespace std; #define INF 2147483647 #define Open(s) freopen(s".in","r",stdin);freopen(s".out","w",stdout); #define Close fclose(stdin); fclose(stdout); const int MAXN = 200010; inline int read(){int s = 0, w = 1;char ch = getchar();while(ch < '0' || ch > '9') { if(ch == '-') w = -1; ch = getchar(); }while(ch >= '0' && ch <= '9') { s = s * 10 + ch - '0'; ch = getchar(); }return s * w; } struct Edge{int next, to, dis; }e[MAXN << 1]; int num, head[MAXN], flag[MAXN]; inline void Add(int from, int to, int dis){e[++num] = (Edge){ head[from], to, dis };head[from] = num; } int pre[MAXN], in[MAXN]; int n, ans, Maxu, ans1; long long Max; void dfs(int u, int fa, long long dep){if(dep > Max) Max = dep, Maxu = u;for(int i = head[u]; i; i = e[i].next)if(e[i].to != fa)dfs(e[i].to, u, dep + e[i].dis); } void DFS(int u, int fa, long long dep){pre[u] = fa;if(dep > Max) Max = dep, Maxu = u;for(int i = head[u]; i; i = e[i].next)if(e[i].to != fa)DFS(e[i].to, u, dep + e[i].dis); } bool existOther(int u, int fa, long long dep){if(dep == Max) return true;for(int i = head[u]; i; i = e[i].next)if(e[i].to != fa)if(existOther(e[i].to, u, dep + e[i].dis)) return true;return false; } int A, B, C; int main(){Open("diameter");n = read();for(int i = 1; i < n; ++i){A = read(); B = read(); C = read();Add(A, B, C); Add(B, A, C);}Max = -1; dfs(1, 0, 0);Max = -1; DFS(Maxu, 0, 0);int now = Maxu, o = 0, fa = 0, Plus = 0;long long deep = 0;while(now) flag[now] = 1, now = pre[now];for(int i = 1; i <= n; ++i) if(!flag[i]) pre[i] = 0;now = Maxu;while(now){Plus = 0;for(int i = head[now]; i; i = e[i].next)if(e[i].to != pre[now] && e[i].to != fa){if(existOther(e[i].to, now, deep + e[i].dis)){o = now;break;}}else if(e[i].to == pre[now]) Plus = e[i].dis;if(!o && !pre[now]) o = now;if(o) break;fa = now;now = pre[now];deep += Plus;} //找到第一個分叉的點ans = 0; //往回遍歷now = o;fa = pre[now];deep = Max - deep;while(now != Maxu){int nxt;++ans;for(int i = head[now]; i; i = e[i].next)if(pre[e[i].to] != now && e[i].to != fa){if(existOther(e[i].to, now, deep + e[i].dis)){printf("%I64d\n%d\n", Max, ans - 1);//system("pause");return 0;}}else if(pre[e[i].to] == now) nxt = e[i].to, Plus = e[i].dis;fa = now;now = nxt;deep += Plus;}printf("%I64d\n%d\n", Max, ans); //如果都不能分叉Close;//system("pause");return 0; }

轉(zhuǎn)載于:https://www.cnblogs.com/Qihoo360/p/9541297.html

總結(jié)

以上是生活随笔為你收集整理的【洛谷 P3304】[SDOI2013]直径（树的直径）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。