hdu4514
题目
首先给你一个图,边有边权,如果能构成一个环就输出yes,否则在森林中找一条最长链。
思路
判断环就用并查集,找最长链就用树形dp,(对于每一个点,访问它的一个子节点,就用这个点的当前最大子链和这个子节点更新结果,再更新这个点的最大子链),一开始题意理解错了,总之也算是懂了最长链的求法233。(边貌似要开的大点,用c++交……)
代码
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
#include <iostream>
#include <stack>
#include <queue>
using namespace std;
const int maxn=1001000;
int pre[maxn];
int ans;
int Find(int x)
{
if(pre[x]!=x) pre[x]=Find(pre[x]);
return pre[x];
}
struct node
{
int next;
int to;
int w;
} edge[2000100];
int head[maxn],tot,dp[maxn];
void addedge(int from,int to,int w)
{
edge[tot].to=to;
edge[tot].next=head[from];
edge[tot].w=w;
head[from]=tot++;
}
int n,m;
void dfs(int u,int fa)
{
int maxx=0;
for(int i=head[u]; ~i; i=edge[i].next)
{
int v=edge[i].to;
int w=edge[i].w;
if(fa==v) continue;
dfs(v,u);
ans=max(ans,dp[v]+w+maxx);
maxx=max(maxx,dp[v]+w);
}
dp[u]=maxx;
}
int main()
{
while(scanf("%d %d",&n,&m)!=EOF)
{
tot=0;
for(int i=0; i<=n; i++) pre[i]=i;
int flag=0;
memset(head,-1,sizeof(head));
for(int i=0; i<m; i++)
{
int a,b,c;
scanf("%d %d %d",&a,&b,&c);
int x=Find(a);
int y=Find(b);
addedge(a,b,c);
addedge(b,a,c);
if(x==y)
{
flag=1;
}
else
{
pre[x]=y;
}
}
if(flag)
{
printf("YES\n");
continue;
}
ans=0;
memset(dp,-1,sizeof(dp));
for(int i=1; i<=n; i++)
{
if(dp[i]==-1)
dfs(i,-1);
}
printf("%d\n",ans);
}
return 0;
}