首页 > 其他分享 >洛谷 P2934 [USACO09JAN] Safe Travel G 题解

洛谷 P2934 [USACO09JAN] Safe Travel G 题解

时间:2024-03-19 16:33:58浏览次数:24  
标签:洛谷 idx int 题解 Travel tree len yzh now

前话

貌似别人都是使用并查集维护的方法,然而由于排序、最短路等算法瓶颈,以下令 \(n\) 和 \(m\) 同阶,总的时间复杂度依然是 \(\Theta(n \log n)\) 的,那么并查集是否有点大材小用了。事实上,在建完最短路径树后,我给出了两种带 \(\log\) 的数据结构完成此题。

题目分析

翻译里已经把问题抽象出来了,这里不过多赘述。考虑到从 \(1\) 到任意结点的最短路是唯一的,说明将所有最短路径保留,删去其它边,一定能形成一棵树,这棵树就是最短路径树。我们称保留下来的边为树边,删去的边为非树边。

题目要求不经过原来最短路上最后一条边的最短路,也就是删去 \(u\) 和 \(fa[u]\) 之间的树边后,通过剩余的树边非树边到达 \(u\) 的最短路。很容易想到,必然要经过一条非树边,而且最多只经过一条非树边。前者考虑连通性证明显然,后者有如下证明:

证明:
假设删去 \(u\) 和 \(fa[u]\) 之间的树边后,经过了两条非树边得到最短路径是 \(1 \stackrel{\color{blue}{\texttt{树边}}}{\color{blue}{\longrightarrow}} xym \stackrel{\color{red}{\texttt{非树边}}}{\color{red}{\longrightarrow}} xym' \stackrel{\color{blue}{\texttt{树边}}}{\color{blue}{\longrightarrow}} yzh \stackrel{\color{red}{\texttt{非树边}}}{\color{red}{\longrightarrow}} yzh' \stackrel{\color{blue}{\texttt{树边}}}{\color{blue}{\longrightarrow}} u\),那么根据最短路径树的相关性质,有路径 \(1 \stackrel{\color{blue}{\texttt{树边}}}{\color{blue}{\longrightarrow}} yzh \stackrel{\color{red}{\texttt{非树边}}}{\color{red}{\longrightarrow}} yzh' \stackrel{\color{blue}{\texttt{树边}}}{\color{blue}{\longrightarrow}} u\) 比以上路径更短,原假设不成立,经过多条非树边情况同理。证毕。

考虑走的过程,容易知道 \(yzh\) 不能是 \(u\) 子树里的节点,因为如果是的话,走到 \(yzh\) 的时候就经过了 \(fa[u] \rightarrow u\) 这条边,与题意不符。另外,\(yzh'\) 是 \(u\) 子树里的结点(当然 \(u\) 也属于 \(u\) 的子树里),这样才能一路向上走走到 \(u\)。放张图吧。

记 \(d_u\) 为 \(u\) 的最短路长度,即 \(1 \rightarrow u\) 这条路径的距离,记 \(\operatorname{dist}(u, v)\) 为 \(u\) 经过一条非树边走到 \(v\) 的长度。

所以先跑一遍最短路建出最短路径树。考虑第一遍深搜,枚举 \(yzh\) 连出的所有非树边 \(yzh \rightarrow yzh'\),在 \(yzh'\) 上打上一个标记 \(d_{yzh} + \operatorname{dist}(yzh, yzh')\),这样在第二遍深搜的时候,就可以把子树的标记不断向上传递、合并,把子树的所有标记加上到父节点的距离,再和父节点合并,就是模拟 \(yzh'\) 到 \(u\) 的过程。然后得到答案的时候就是取出标记里最小的那个就行了。但是发现这样存在一个问题,就是不能保证不会出现上文提到的 \(yzh\) 不能是 \(u\) 子树里的节点这个要求,也就是在不断向上传递的时候,有可能传递到了 \(yzh\) 甚至 \(yzh\) 的祖先,这显然会造成问题。解决方法就是将标记增加一个信息变为 \((d_{yzh} + \operatorname{dist}(yzh, yzh'), yzh)\),这样每次取出当前最小值时,发现 \(yzh\) 不满足要求,就将这个标记删除。判断合法可以在第一遍深搜时处理出 dfs 序判断。这个过程套路化地有以下两种方法解决。

1. 左偏树 + 懒惰标记

合并、整体加、删除最小值、获取最小值,明显是可合并堆的范畴啊,这里使用左偏树解决。顺带一提,由于有整体加这个操作,不好使用启发式合并,所以老老实实写左偏树吧

2. 线段树

什么?你竟然不会使用左偏树,那就快乐地打线段树吧!

为什么要合并?我们只用知道整个子树的信息啊,所以可以在另外记一个信息的 dfs 序,然后树上问题有变成序列上的问题了,区间加、求最小值,使用线段树维护,删除的时候将其赋成 \(\infty\) 就能不会对之后产生影响。注意区分两个 dfs 序的区别。

代码(已略去快读快写,码风清新,注释详尽)

1. 左偏树 + 懒惰标记

//#pragma GCC optimize(3)
//#pragma GCC optimize("Ofast", "inline", "-ffast-math")
//#pragma GCC target("avx", "sse2", "sse3", "sse4", "mmx")
#include <iostream>
#include <cstdio>
#define debug(a) cerr << "Line: " << __LINE__ << " " << #a << endl
#define print(a) cerr << #a << "=" << (a) << endl
#define file(a) freopen(#a".in", "r", stdin), freopen(#a".out", "w", stdout)
#define main Main(); signed main(){ return ios::sync_with_stdio(0), cin.tie(0), Main(); } signed Main
using namespace std;

#include <cstring>
#include <queue>

int n, m;
int ans[100010];

struct Graph{
	struct node{
		int to, len, nxt;
	} edge[200010 << 1];
	int eid, head[100010];
	inline void add(int u, int v, int w){
		edge[++eid] = {v, w, head[u]};
		head[u] = eid;
	}
	node & operator [] (const int x){
		return edge[x];
	}
} xym, yzh;
// 一个是原图,一个是最短路径树

// 最短路,父亲,和父亲的距离
int dist[100010], fr[100010], lenfa[100010];
template <class T> using MinHeap = priority_queue<T, vector<T>, greater<T> >;

// dfs 序判断子树
int L[100010], R[100010], timer;

int root[100010];  // u 在左偏树里的结点
int fa[400010], dis[400010], lson[400010], rson[400010];  // 左偏树
int lazy[400010];  // 左偏树懒惰标记
pair<int, int> val[400010];  // 信息
int pcnt;

int NewNode(pair<int, int> v){
	return val[++pcnt] = v, fa[pcnt] = pcnt;
}

void pushtag(int u, int tag){
	val[u].first += tag;
	lazy[u] += tag;
}

void pushdown(int u){
	if (lazy[u] == 0) return;
	if (lson[u]) pushtag(lson[u], lazy[u]);
	if (rson[u]) pushtag(rson[u], lazy[u]);
	lazy[u] = 0;
}

int combine(int a, int b){
	if (!a || !b) return a | b;
	if (val[a] > val[b]) swap(a, b);
	pushdown(a), rson[a] = combine(rson[a], b);
	if (dis[lson[a]] < dis[rson[a]]) swap(lson[a], rson[a]);
	dis[a] = dis[rson[a]] + 1;
	return a;
}

// 第一遍 dfs
void dfs(int now){
	L[now] = ++timer;  // 记录 dfs 序
	for (int i = xym.head[now]; i; i = xym[i].nxt){  // 枚举 yzh -> yzh' 这条非树边
		int to = xym[i].to, len = xym[i].len;
		if (fr[to] == now && len == lenfa[to]) continue;  // 这里要很小心地判断树边
		int node = NewNode({dist[now] + len, now});
		if (!root[to]) root[to] = node;
		else root[to] = combine(root[to], node);
	}
	for (int i = yzh.head[now]; i; i = yzh[i].nxt) dfs(yzh[i].to);
	R[now] = timer;
}

// 第二遍 dfs
void redfs(int now){
	for (int i = yzh.head[now]; i; i = yzh[i].nxt){  // 把子树标记合并过来
		int to = yzh[i].to, len = yzh[i].len;
		redfs(to);
		if (!root[to]) continue;
		pushtag(root[to], len);
		if (!root[now]) root[now] = root[to];
		else root[now] = combine(root[now], root[to]);
	}
	// 弹出不符合的标记
	while (root[now] && L[now] <= L[val[root[now]].second] && L[val[root[now]].second] <= R[now]){
		if (lson[root[now]] == 0 && rson[root[now]] == 0){
			root[now] = 0;
		} else {
			root[now] = combine(lson[root[now]], rson[root[now]]);
		}
	}
	// 统计答案
	if (root[now]) ans[now] = val[root[now]].first;
	else ans[now] = -1;
}

// 迪迦哥斯拉算法(逃
// 迪杰斯特拉求最短路,并建出最短路径树
void build(){
	memset(dist, 0x3f, sizeof dist);
	MinHeap<pair<int, int> > Q; Q.push({dist[1] = 0, 1});
	while (!Q.empty()){
		auto [ndis, now] = Q.top(); Q.pop();
		if (dist[now] < ndis) continue;
		for (int i = xym.head[now]; i; i = xym[i].nxt){
			int to = xym[i].to, len = xym[i].len;
			if (dist[to] > dist[now] + len){
				Q.push({dist[to] = dist[now] + len, to});
				fr[to] = now, lenfa[to] = len;
			}
		}
	}
	for (int i = 2; i <= n; ++i) yzh.add(fr[i], i, lenfa[i]);
}

signed main(){
	read(n, m);
	for (int i = 1, u, v, w; i <= m; ++i) read(u, v, w), xym.add(u, v, w), xym.add(v, u, w);
	build(), dfs(1), redfs(1);
	for (int i = 2; i <= n; ++i) write(ans[i], '\n');
	return 0;
}

2. 线段树

//#pragma GCC optimize(3)
//#pragma GCC optimize("Ofast", "inline", "-ffast-math")
//#pragma GCC target("avx", "sse2", "sse3", "sse4", "mmx")
#include <iostream>
#include <cstdio>
#define debug(a) cerr << "Line: " << __LINE__ << " " << #a << endl
#define print(a) cerr << #a << "=" << (a) << endl
#define file(a) freopen(#a".in", "r", stdin), freopen(#a".out", "w", stdout)
#define main Main(); signed main(){ return ios::sync_with_stdio(0), cin.tie(0), Main(); } signed Main
using namespace std;

#include <cstring>
#include <queue>
#include <vector>

const int inf = 0x3f3f3f3f;

int n, m;
int ans[100010];

struct Graph{
	struct node{
		int to, len, nxt;
	} edge[200010 << 1];
	int eid, head[100010];
	inline void add(int u, int v, int w){
		edge[++eid] = {v, w, head[u]};
		head[u] = eid;
	}
	node & operator [] (const int x){
		return edge[x];
	}
} xym, yzh;
// 一个是原图,一个是最短路径树

// 最短路,父亲,和父亲的距离
int dist[100010], fr[100010], lenfa[100010];
template <class T> using MinHeap = priority_queue<T, vector<T>, greater<T> >;

// dfs 序判断子树
int L[100010], R[100010], timer;

// 信息的 dfs 序
int infoL[100010], infoR[100010], infot;
vector<pair<int, int> > info[100010];
pair<int, int> val[400010];  // 注意这里要开大一些

struct Segment_Tree{
	#define lson (idx << 1    )
	#define rson (idx << 1 | 1)
	
	struct Info{
		pair<int, int> val;
		int pos;
		Info operator + (const Info & o) const {
			if (val.first == inf) return o;
			if (o.val.first == inf) return *this;
			if (val.first < o.val.first) return *this;
			return o;
		}
		Info operator + (const int o) const {
			if (val.first == inf) return *this;
			return {{val.first + o, val.second}, pos};
		}
	};
	// 信息
	
	struct node{
		int l, r;
		Info info;
		int tag;
	} tree[400010 << 2];
	
	void pushup(int idx){
		tree[idx].info = tree[lson].info + tree[rson].info;
	}
	
	void build(int idx, int l, int r){
		tree[idx] = {l, r, {{inf, -1}, -1}, 0};
		if (l == r) return tree[idx].info = {val[l], l}, void();
		int mid = (l + r) >> 1;
		build(lson, l, mid), build(rson, mid + 1, r), pushup(idx);
	}
	
	void pushtag(int idx, const int t){
		tree[idx].info = tree[idx].info + t;
		tree[idx].tag = tree[idx].tag + t;
	}
	
	void pushdown(int idx){
		if (!tree[idx].tag) return;
		pushtag(lson, tree[idx].tag), pushtag(rson, tree[idx].tag);
		tree[idx].tag = 0;
	}
	
	Info query(int idx, int l, int r){
		if (tree[idx].l > r || tree[idx].r < l) return {{inf, -1}, -1};
		if (l <= tree[idx].l && tree[idx].r <= r) return tree[idx].info;
		return pushdown(idx), query(lson, l, r) + query(rson, l, r);
	}
	
	void modify(int idx, int l, int r, const int t){
		if (tree[idx].l > r || tree[idx].r < l) return;
		if (l <= tree[idx].l && tree[idx].r <= r) return pushtag(idx, t);
		pushdown(idx), modify(lson, l, r, t), modify(rson, l, r, t), pushup(idx);
	}
	
	void erase(int idx, int p){
		if (tree[idx].l > p || tree[idx].r < p) return;
		if (tree[idx].l == tree[idx].r) return tree[idx].info = {{inf, -1}, -1}, void();
		pushdown(idx), erase(lson, p), erase(rson, p), pushup(idx);
	}
	
	#undef lson
	#undef rson
} tree;

// 第一遍 dfs
void dfs(int now){
	L[now] = ++timer;  // 记录 dfs 序
	for (int i = xym.head[now]; i; i = xym[i].nxt){  // 枚举 yzh -> yzh' 这条非树边
		int to = xym[i].to, len = xym[i].len;
		if (fr[to] == now && len == lenfa[to]) continue;  // 这里要很小心地判断树边
		info[to].push_back({dist[now] + len, now});
	}
	for (int i = yzh.head[now]; i; i = yzh[i].nxt) dfs(yzh[i].to);
	R[now] = timer;
}

// 其实也算一次深搜,记录标记的 dfs 序
void prework(int now){
	infoL[now] = infot + 1;
	for (auto x: info[now]) val[++infot] = x;
	for (int i = yzh.head[now]; i; i = yzh[i].nxt) prework(yzh[i].to);
	infoR[now] = infot;
}

// 第二遍 dfs
void redfs(int now){
	if (infoL[now] > infoR[now]) return;
	for (int i = yzh.head[now]; i; i = yzh[i].nxt){  // 把子树标记合并过来
		int to = yzh[i].to, len = yzh[i].len;
		redfs(to), tree.modify(1, infoL[to], infoR[to], len);
	}
	// 弹出不符合的标记
	while (true){
		Segment_Tree::Info res = tree.query(1, infoL[now], infoR[now]);
		if (res.val.first == inf || res.pos == -1) break;
		if (L[now] <= L[res.val.second] && L[res.val.second] <= R[now]){
			tree.erase(1, res.pos);
			continue;
		}
		// 统计答案
		ans[now] = res.val.first;
		break;
	}
}

// 迪迦哥斯拉算法(逃
// 迪杰斯特拉求最短路,并建出最短路径树
void build(){
	memset(dist, 0x3f, sizeof dist);
	MinHeap<pair<int, int> > Q; Q.push({dist[1] = 0, 1});
	while (!Q.empty()){
		auto [ndis, now] = Q.top(); Q.pop();
		if (dist[now] < ndis) continue;
		for (int i = xym.head[now]; i; i = xym[i].nxt){
			int to = xym[i].to, len = xym[i].len;
			if (dist[to] > dist[now] + len){
				Q.push({dist[to] = dist[now] + len, to});
				fr[to] = now, lenfa[to] = len;
			}
		}
	}
	for (int i = 2; i <= n; ++i) yzh.add(fr[i], i, lenfa[i]);
}

signed main(){
	read(n, m);
	for (int i = 1, u, v, w; i <= m; ++i) read(u, v, w), xym.add(u, v, w), xym.add(v, u, w);
	for (int i = 2; i <= n; ++i) ans[i] = -1;
	build(), dfs(1), prework(1), tree.build(1, 1, infot), redfs(1);
	for (int i = 2; i <= n; ++i) write(ans[i], '\n');
	return 0;
}

总结 & 后话

考场上没想到可以将长度拆开来算,也就是计算答案的时候不用一步一步加与父亲的长度加上去,而是用 \(d_{yzh'} - d_u\) 计算,这样对于 \(u\) 来说,\(d_u\) 是确定的,要求的就是 \(\min \lbrace d_{yzh} + \operatorname{dist}(yzh, yzh') + d_{yzh'} \rbrace\),这样可以愉快地用启发式合并秒了啊!但是练习一道左偏树懒惰标记的题目,感觉也颇有收获呢。但是但是,线段树无敌爱敲

标签:洛谷,idx,int,题解,Travel,tree,len,yzh,now
From: https://www.cnblogs.com/XuYueming/p/18083293

相关文章

  • 腾讯春招内参:2024最全Spring Boot面试题解析,技术精英必备!
    随着2024年春季招聘季的来临,腾讯再次开启了对富有才华和创新精神的技术人才的寻找之旅。作为一家全球领先的互联网科技公司,腾讯在寻找那些不仅拥有扎实的技术基础,而且能够适应快速发展和变化的行业环境的候选人。在众多技术栈中,SpringBoot作为简化Spring应用开发的工具,因其......
  • AtCoder Beginner Contest 345 A-F 题解
    A-LeftrightarrowQuestion给你一个由<、=和>组成的字符串\(S\)。请判断\(S\)是否是双向箭头字符串。字符串\(S\)是双向箭头字符串,当且仅当存在一个正整数\(k\),使得\(S\)是一个<、\(k\)个=和一个>的连接,且顺序如此,长度为\((k+2)\)。Solution按照题意......
  • 分月饼【华为OD机试JAVA&Python&C++&JS题解】
    一.题目-分月饼中秋节,公司分月饼,m个员工,买了n个月饼,m<=n,每个员工至少分1个月饼,但可以分多个,单人分到最多月饼的个数是Max1,单人分到第二多月饼个数是Max2,Max1-Max2<=3,单人分到第n-1多月饼个数是Max(n-1),单人分到第n多月饼个数是Max(n),Max(n-1)–Max(n)<=3,问有多少......
  • ARC174C 题解
    blog。官解似乎很难想到,这里是容易想到的方法。显然是DP。介于轮数可能趋近于无穷,所以类似P4550做即可。设\(f_i,g_i\)表示已经抽了\(i\)个数,当前是Alice或Bob抽的,期望罚款。倒推处理,\(f_n=g_n=0\)。下文中\(p=\dfracin\)表示抽到已经有的概率。\[\large\begin......
  • [极客大挑战 2019]web部分题解(sql部分已完结,其他部分正在更新)
    SQL部分:[极客大挑战2019]BabySQL打开环境后有登录界面◕‿◕一眼注入,后先试试万能密码:username:admin'or'1'='1password:1 GG,出大问题,我就会这一招啊O.o??完结撒花(不是꒰ঌ(⌯''⌯)໒꒱开玩笑的,着看着像是过滤了or后来尝试了一下oorr双写发现也不行,那咱继续注入哈:尝试......
  • 洛谷题单指南-二叉树-P1185 绘制二叉树
    原题链接:https://www.luogu.com.cn/problem/P1185题意解读:在表格中绘制二叉树,有几个关键点1、结点用小写字母o 表示,对于一个父亲结点,用 / 连接左子树,用 \连接右子树,表格其余地方填空格。2、第m层结点若两个属于同一个父亲,那么它们之间由 3个空格隔开;若两个结点相邻但......
  • 题解:CF1941G Rudolf and Subway
    原题链接简化题意一个无向连通图中将边分成了不同颜色(保证同种颜色联通),问从\(b\)到\(e\)最短需要经过几种颜色思路考虑因为同种颜色联通,可直接在读入的时候开两个vector分别存每个点属于的颜色及每种颜色有哪些点,又因为颜色数字可能跨度比较大,最好另开一个存颜色的种......
  • P9077 [PA2018] Poddrzewo 题解
    思考感觉题目有点迷惑的意思,要最小化操作\(1\)使用的次数,也就是要节约修改操作,让我们认为操作\(1\)是最有用的,其实只要稍微动动脑子想一想,删除操作才是最有用的,而交换操作根本没用。当将序列删除到只剩两个点时,就把两个点连上,度都为\(1\)。所以如果序列中\(1\)的数量超......
  • 亲子游戏【华为OD机试JAVA&Python&C++&JS题解】
    题目描述宝宝和妈妈参加亲子游戏,在一个二维矩阵(NN)的格子地图上,宝宝和妈妈抽签决定各自的位置,地图上每个格子有不同的糖果数量,部分格子有障碍物。游戏规则是妈妈必须在最短的时间(每个单位时间只能走一步)到达宝宝的位置,路上的所有糖果都可以拿走,不能走障碍物的格子,只能上下......
  • 洛谷-P1449 后缀表达式
    目录 何为后缀表达式?模拟过程AC代码采用STL的stack题目链接:P1449后缀表达式-洛谷|计算机科学教育新生态(luogu.com.cn) 何为后缀表达式?那后缀表达式是怎么算的呢那显然就需要引用最开始说的栈了因为后缀表表达式本来就是栈的一种应用那么现在来说说后缀表......