hdu4514 湫湫系列故事——设计风景线(并查集+树形dp)

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;
}