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

运行结果:

相关推荐
freexyn3 分钟前
Matlab自学笔记五十一:(推荐)输入参数的数量和可变数量的输入
笔记·算法·matlab
陈大大陈6 分钟前
基于 C++ 的用户认证系统开发:从注册登录到Redis 缓存优化
java·linux·开发语言·数据结构·c++·算法·缓存
纪元A梦8 分钟前
华为OD机试真题——通过软盘拷贝文件(2025A卷:200分)Java/python/JavaScript/C++/C语言/GO六种最佳实现
java·javascript·c++·python·华为od·go·华为od机试题
数据分析螺丝钉23 分钟前
LeetCode 252 会议室 III(Meeting Rooms III)题解与模拟面试
算法·leetcode·职场和发展
小李独爱秋26 分钟前
动态哈希映射深度指南:从基础到高阶实现与优化
数据结构·算法·哈希算法
猫猫头有亿点炸36 分钟前
C语言斐波拉契数列2.0
c语言·开发语言·算法
刚入坑的新人编程43 分钟前
C++多态
开发语言·c++
写个博客1 小时前
代码随想录算法训练营第二十六天
算法
QUST-Learn3D1 小时前
高精度并行2D圆弧拟合(C++)
开发语言·c++
明月醉窗台1 小时前
Qt 入门 6 之布局管理
c语言·开发语言·c++·qt