湫湫系列故事——设计风景线(并查集) [2013腾讯编程马拉松初赛第二场(3月22日)]

原题:http://acm.hdu.edu.cn/showproblem.php?pid=4514

还是练得太少,去年准备复试时,每每遇到连通性问题,刚开始总是想到是需要实现图的表示,再遍历求解,后来看到牛人都是一个并查集便可轻松解决;

还是平时练得少,学习即使练,也没有高度的集中精力,往后的学习也应该有这种高度集中解决问题的专注力。

该题解法要点: 

     1)构建并查集,若发现新加入的两个节点共享一个根节点,则有环;

     2)每合并两个节点node1、node2,则新的根节点sum = node1.sum + node2.sum + weight(node1, node2)

#include <iostream>
#include <algorithm>
using namespace std;
class BCJ {
public:
    int next;
    int sum;
    bool operator < (BCJ b) {
        return this->sum < b.sum;
    }
};
BCJ bcj[100001];
BCJ tbcj = {-1, 0};

int find(int index)
{
    while(bcj[index].next >= 0)
        index = bcj[index].next;
    return index;
}

void merge(int a , int b, int weight)
{
    a = find(a);
    b = find(b);
    bcj[a].next =  b;
    bcj[b].sum += bcj[a].sum + weight;
}
int main() {
    int n, m, start, end, weight, i, root1, root2;
    bool hasCircle;
    while(cin>>n>>m) {
        //init
        hasCircle = false;
        fill_n( bcj, n + 1, tbcj);

        for(i = 0 ; i < m ; i++) {
            cin>> start >> end >> weight;
            root1  = find(start);
            root2  = find( end);
            if(root1 != root2) {
                merge(root1, root2, weight);
            } else {
                hasCircle = true;
            }
        }
        if( hasCircle) {
            cout << "YES" <<endl;
        } else {
            cout << max_element(bcj + 1 , bcj + n + 1)->sum << endl;
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值