新春漫步（并查集）---“迎新春，过大年”多校程序设计竞赛-G

最新推荐文章于 2024-10-01 23:05:47 发布

陈进士学习

最新推荐文章于 2024-10-01 23:05:47 发布

阅读量396

点赞数 9

分类专栏：牛客文章标签：算法 c++ 开发语言 c语言并查集

本文链接：https://blog.csdn.net/JungleZRD/article/details/136247299

版权

牛客专栏收录该内容

170 篇文章 1 订阅

订阅专栏

在这里插入图片描述

解析

并查集，将其终点作为并查集的根节点

注意特判结尾 $n$ 的情况

//G
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define inf 0x3f3f3f3f
const int N=1e6+5;
int n,m;
int a[N],p[N];
int find(int x){
	return x==p[x]?x:p[x]=find(p[x]);
}
void solve(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		p[i]=i;
	}
	while(m--){
		int op,x,y;
		cin>>op;
		if(op==1){
			cin>>x>>y;
			a[x]-=y;
			if(a[x]<=0){
				if(x<n){
					p[x]=find(x+1);
				}
			}
		}
		else{
			cin>>x;
			cout<<find(x)<<endl;
		}
	}
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	int t=1;
//	cin>>t;
	while(t--) solve();
	return 0;
}