D - Hidden Weights

题目链接:

D - Hidden Weights (atcoder.jp)

题目描述:

数据范围限制:

样例:

输入:

3 3

1 2 2

3 2 3

1 3 -1

输出:

3 5 2

样例图:

思路:

第一次想的是从没有入节点的点去出发,然后去依次去更新,但是发现这个是错误的。

dfs去暴力搜索,也就是去遍历点。遍历过的点就标记为true,遍历到当前这个点的时候,这个点的后面的点也就会遍历,因此没有必要去重复标记。

注意:在加入点的时候 add(x, y, z)add(y, x, -z) 因为 x到y的距离是z,那么反过来就是-z 。为什么要加第二个add呢?要是不加的话dfs() 的时候标记为true,那么后面再去由其他的点遍历到当前这个点的时候,发现为true就不能再去更新了。例如下面这张图,要是不去add(y, x, -z) 的话, 那么在for(int i = 1; i <= n; i ++ ) i=1的时候发现ok[1] = false 然后去dfs,把其标记为true, 然后for的时候发现它没有指向别的边,当遍历到2的时候,发现2的下面有个1,尝试着去更新1的时候,发现1已经被标记为true了,怎么去更新?要是加入add(y, x, -z) 的话就相当于是个双向图了,但是权值是相反的,不过也没事,每个点遵循的原则是:只去更新一次。

这个add两次的这个方法在之前是见过的,但是没想到,我用的是类似与bfs的方法,但是这样是会发生冲突的,用dfs这个不撞南墙不回头的精神,因为答案必存在,那么任何一个起点都是可以的,不会发生冲突。

代码:

cpp 复制代码
#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
 
using namespace std;

const int N = 2e5 + 10;

vector<pair<int, int> > v[N];
int n, m;
int ret[N];
int ok[N];

// 这样确实会一个点去遍历一次 
void dfs(int x) {
	cout << "x = " << x << endl ;
	ok[x] = true;
	for(auto i : v[x]) {
		//cout << "i = " << i.fi << endl;
		if(!ok[i.fi]) { // 还没有被更新 
			ret[i.fi] = ret[x] + i.se;
		//	cout << "i.fi = " << i.fi << endl;
			dfs(i.fi);;
		}
	}
}

signed main() {
	cin >> n >> m;
	for(int i = 1; i <= m; i ++ ) {
		int x, y, z;
		cin >> x >> y >> z;
		v[x].push_back({y, z});
		v[y].push_back({x, -z}); // 这个是必须要写的
		// 解决了先后的问题 
	}
	
	for(int i = 1; i <= n; i ++ ) {
		if(!ok[i])
		dfs(i);
	}
	
	for(int i = 1; i <= n; i ++ ) {
		cout << ret[i] << " ";
	}
	
	return 0;
}

运行结果:

相关推荐
机器视觉知识推荐、就业指导9 分钟前
C++设计模式:建造者模式(Builder) 房屋建造案例
c++
Swift社区1 小时前
LeetCode - #139 单词拆分
算法·leetcode·职场和发展
Kent_J_Truman1 小时前
greater<>() 、less<>()及运算符 < 重载在排序和堆中的使用
算法
IT 青年2 小时前
数据结构 (1)基本概念和术语
数据结构·算法
Yang.992 小时前
基于Windows系统用C++做一个点名工具
c++·windows·sql·visual studio code·sqlite3
熬夜学编程的小王2 小时前
【初阶数据结构篇】双向链表的实现(赋源码)
数据结构·c++·链表·双向链表
zz40_2 小时前
C++自己写类 和 运算符重载函数
c++
Dong雨2 小时前
力扣hot100-->栈/单调栈
算法·leetcode·职场和发展
六月的翅膀2 小时前
C++:实例访问静态成员函数和类访问静态成员函数有什么区别
开发语言·c++