Forsaken喜欢独一无二的树

Forsaken喜欢独一无二的树

https://ac.nowcoder.com/acm/problem/53074

题意:

图片说明

分析:

正如“jxnu-19-软技一班-刘晟”所说的,按照kruskai算法的想法来看的话,出现多条最短路径的原因只可能是存在复数的最小权值边连接着两个集合以供选择。
那我们其实只要在kruskai算法进行的同时动手脚就行了。
我们先对权值进行从小到大的排序,然后将其分为一个个的相等权值的集合。
对每一份的相等权值集合,我们看这里面有没有复数的连接两个集合的边,有的话我们实际上只需连一条,所以剩下的条数*权值就是我们要消去的。

代码:

#include<iostream>
#include<algorithm>
#include<vector>
#include<set>
using namespace std;
typedef long long ll;
const int max_n = 2e5 + 100;
typedef pair<int, int> pii;
struct edge
{
    int u, v, cost;
    bool operator<(const edge& e) { return cost < e.cost; }
}E[max_n];
int n, m;

int par[max_n];
int rak[max_n];

void init(int n) {
    for (int i = 1;i <= n;i++)par[i] = i;
    fill(rak, rak + 1 + n, 0);
}

int find(int x) {
    if (par[x] == x)return x;
    else return par[x] = find(par[x]);
}

bool same(int x, int y) {
    return find(x) == find(y);
}

bool merge(int x, int y) {
    if ((x = find(x)) == (y = find(y)))return false;
    if (rak[x] > rak[y])par[y] = x;
    else if (rak[x] < rak[y])par[x] = y;
    else par[y] = x, rak[x]++;
    return true;
}

ll kruskai_change() {
    ll ans = 0;init(n);
    sort(E + 1, E + 1 + m);
    int i = 1;int j = 1;
    while (i <= m) {
        while ((j <= m) && (E[j].cost == E[i].cost))j++;
        for (int k = i;k < j;k++) {
            int u = E[k].u;int v = E[k].v;
            if (!same(u, v))ans += E[k].cost;
        }
        for (int k = i;k < j;k++) {
            int u = E[k].u;int v = E[k].v;
            if (!same(u, v))ans -= E[k].cost, merge(u, v);
        }
        i = j;
    }return ans;
}

int main() {
    ios::sync_with_stdio(0);
    cin >> n >> m;
    for (int i = 1;i <= m;i++)
        cin >> E[i].u >> E[i].v >> E[i].cost;
    cout << kruskai_change() << endl;
}

本题双重遍历的代码小技巧,学到了。欸嘿~

全部评论

相关推荐

点赞 评论 收藏
分享
Southyeung:我说一下我的看法(有冒犯实属抱歉):(1)简历不太美观,给我一种看都不想看的感觉,感觉字体还是排版问题;(2)numpy就一个基础包,机器学习算法是什么鬼?我感觉你把svm那些写上去都要好一点。(2)课程不要写,没人看,换成获奖经历;(3)项目太少了,至少2-3个,是在不行把网上学习的也写上去。
点赞 评论 收藏
分享
05-11 11:48
河南大学 Java
程序员牛肉:我是26届的双非。目前有两段实习经历,大三上去的美团,现在来字节了,做的是国际电商的营销业务。希望我的经历对你有用。 1.好好做你的CSDN,最好是直接转微信公众号。因为这本质上是一个很好的展示自己技术热情的证据。我当时也是烂大街项目(网盘+鱼皮的一个项目)+零实习去面试美团,但是当时我的CSDN阅读量超百万,微信公众号阅读量40万。面试的时候面试官就告诉我说觉得我对技术挺有激情的。可以看看我主页的美团面试面经。 因此花点时间好好做这个知识分享,最好是单拉出来搞一个板块。各大公司都极其看中知识落地的能力。 可以看看我的简历对于博客的描述。这个帖子里面有:https://www.nowcoder.com/discuss/745348200596324352?sourceSSR=users 2.实习经历有一些东西删除了,目前看来你的产出其实很少。有些内容其实很扯淡,最好不要保留。有一些点你可能觉得很牛逼,但是面试官眼里是减分的。 你还能负责数据库表的设计?这个公司得垃圾成啥样子,才能让一个实习生介入数据库表的设计,不要写这种东西。 一个公司的财务审批系统应该是很稳定的吧?为什么你去了才有RBAC权限设计?那这个公司之前是怎么处理权限分离的?这些东西看着都有点扯淡了。 还有就是使用Redis实现轻量级的消息队列?那为什么这一块不使用专业的MQ呢?为什么要使用redis,这些一定要清楚, 就目前看来,其实你的这个实习技术还不错。不要太焦虑。就是有一些内容有点虚了。可以考虑从PR中再投一点产出
投递美团等公司9个岗位
点赞 评论 收藏
分享
评论
3
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务