删边
Description
給出N個點,N-1條邊的連通圖.
現要求刪除一條邊,使得連通塊的直徑總和最大.所謂連通塊的直徑是指連通塊中最遠兩點之間的距離。
問:直徑總和最大是多少?
Input
文件名為 delete.in
第一行正整數N.
接下來N-1行.每行兩個數,A,B,LEN表示A,B(1<=A,B<=N)有一條長度為Len(1<=Len<=1000)的邊連接著.
Output
文件名為 delete.out
一個數Ans.直徑總和的最大值.
Sample Input
10
2 1 982
3 1 169
4 1 934
5 1 325
6 1 735
7 1 675
8 2 302
9 3 450
10 5 173
Sample Output
2668
Data Constraint
Hint
【數據范圍】
30% N<=100
70% N<=5000
100% N<=100000
.
.
.
.
.
.
分析
我們要預處理出每一棵子樹的直徑和子樹中的最遠點,次遠點和子樹中過x的最長鏈、次長鏈和次次長鏈
那么考慮刪去一條邊后直徑有哪幾種情況
①在x的子樹里
②在x上面的聯通塊的直徑
③x子樹沒被刪去的最遠點與x上方最遠點的和
④在x不同子樹上最遠的點的和
考慮一下怎么求:
①預處理得出
②在遞歸時可以記錄當前經過聯通塊的最大值
③x子樹里的在預處理里已經求出來,那么不在x子樹里的也就是遞歸經過x的最長鏈
④也就是在x子樹里不含被刪子樹的最遠點和次遠點
.
.
.
.
.
程序:
#include<iostream> using namespace std; int m[2][100001][4],d[2][100001][4],fa[100001],head[100001],dis[100001],ans,n,cnt;struct edge { int to,from,v; }rp[200001];void insert(int x,int y,int z) { rp[++cnt].to=y;rp[cnt].v=z;rp[cnt].from=head[x];head[x]=cnt; }void dfs(int x) {for (int i=head[x];i;i=rp[i].from){int w=rp[i].to;if (w!=fa[x]){fa[w]=x;dfs(w);dis[x]=max(dis[x],dis[w]);if (dis[w]>m[0][x][1]){m[0][x][2]=m[0][x][1];m[0][x][1]=dis[w];d[0][x][1]=w;} else if (dis[w]>m[0][x][2]) m[0][x][2]=dis[w];int v=m[1][w][1]+rp[i].v;if (v>m[1][x][1]){m[1][x][3]=m[1][x][2];m[1][x][2]=m[1][x][1];m[1][x][1]=v;d[1][x][2]=d[1][x][1];d[1][x][1]=w;} elseif (v>m[1][x][2]){m[1][x][3]=m[1][x][2];m[1][x][2]=v;d[1][x][2]=w;} else if (v>m[1][x][3]) m[1][x][3]=v;}}dis[x]=max(m[1][x][1]+m[1][x][2],m[0][x][1]); }void find(int x,int a,int b) {int m1,m2,m3;if (x!=1) ans=max(ans,dis[x]+a);for (int i=head[x];i;i=rp[i].from){int v=rp[i].to;if (v!=fa[x]){if (d[1][x][1]==v){m1=m[1][x][2];m2=m[1][x][2]+m[1][x][3];} else{m1=m[1][x][1];if (d[1][x][2]==v) m2=m[1][x][1]+m[1][x][3]; else m2=m[1][x][1]+m[1][x][2];}if (d[0][x][1]==v) m3=m[0][x][2]; else m3=m[0][x][1];find(v,max(max(a,m3),max(m1+b,m2)),max(b,m1)+rp[i].v);}} } int main() {cin>>n;for (int i=1;i<=n-1;i++){int x,y,z;cin>>x>>y>>z;insert(x,y,z); insert(y,x,z);}dfs(1);find(1,0,0);cout<<ans;return 0; }轉載于:https://www.cnblogs.com/YYC-0304/p/9499936.html
總結
- 上一篇: 【NOI2002】贪吃的九头龙
- 下一篇: 体育场[带权并查集]