牛客练习赛 93 D （带权并查集+珂朵莉树）

最新推荐文章于 2024-10-16 00:00:43 发布

dplovetree

最新推荐文章于 2024-10-16 00:00:43 发布

阅读量318

点赞数 1

分类专栏：数据结构文章标签：算法

本文链接：https://blog.csdn.net/qq_52383696/article/details/121884079

版权

数据结构专栏收录该内容

18 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用带权并查集和珂朵莉树来解决区间操作的问题，具体包括区间赋值和查询。在给定的代码中，实现了对区间进行增援兵力和查询兵力的处理，并通过剪枝优化了时间复杂度。文章还探讨了其他可能的解决方案，如线段树和分块加标记，并鼓励读者尝试不同的方法来解决此类问题。

摘要由CSDN通过智能技术生成

牛客练习赛 93 D

题意：
请添加图片描述

带权并查集+珂朵莉树。
可以发现，对于第一次操作就是类似，删除原本的这一段区间，再把这一段区间赋一个值。这个操作用珂朵莉树可以很好的实现，毕竟号称区间赋值小能手。
给区间设定四个状态，分别是区间左端点、区间右端点、区间每个位置的兵力、区间所属的单位。
对于第一个操作，我们遍历原先在 $l$ ~ $r$ 内的区间，把原本区间属于的祖先减去这个区间的兵力，再让当前的 $i d$ 加上这段增援后的兵力。对于占领的情况，只需要在遍历的时候，判断区间的祖先的家是否在 $l$ ~ $r$ 内，如果在的话，就用带权并查集并到当前的 $i d$ 上，这样就等于把地盘的兵力都给了 $i d$ 。
对于第二个操作，只需要输出 $i d$ 的权值即可。
代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long

int n,m,t;
int fa[100050];
ll w[100050];
int vis[100050];
int pos[100050];

struct node{
	int l,r;
	mutable ll val,x;
	bool operator < (const node &ths)const{
		return l<ths.l;
	}
	node(int L,int R,ll Val,ll X):l(L),r(R),val(Val),x(X){}
	node(int L):l(L){}
};

set<node>s;
using si=set<node>::iterator;

si split(int pos){
	si it=s.lower_bound(node(pos));
	if(it!=s.end()&&it->l==pos)return it;
	--it;
	int l=it->l,r=it->r;
	ll val=it->val;
	ll x=it->x;
	s.erase(it);
	s.insert(node(l,pos-1,val,x));
	return s.insert(node(pos,r,val,x)).first;
}

int find(int x){
	if(x==fa[x])return x;
	return fa[x]=find(fa[x]);
}
vector<int>res;
void solve(int id,int l,int r,int p){
	ll len=r-l+1;
	res.clear();
	si itr=split(r+1),itl=split(l);
	ll cnt=0;
	for(si it=itl;it!=itr;++it){
		int f=find(it->x);
		w[f]-=1LL*(it->r-it->l+1)*(it->val);
		if(f==id)cnt-=1LL*(it->r-it->l+1)*(it->val);
		else {
			cnt+=1LL*(it->r-it->l+1)*(it->val);
			if(pos[f]>=it->l&&pos[f]<=it->r)res.push_back(f);
		}
	}
	printf("%lld\n",1LL*len*p+cnt);
	s.erase(itl,itr);
	s.insert(node(l,r,p,id));
	w[id]+=1LL*(r-l+1)*p;
	for(auto to:res){
		fa[to]=id;
		w[id]+=w[to];
	}
}
int main(){
	scanf("%d%d%d",&n,&m,&t);
	for(int i=1;i<=n;i++){
		scanf("%d",&pos[i]);
		vis[pos[i]]=i;
		fa[i]=i;
		w[i]=0;
	}
	for(int i=1;i<=m;i++){
		s.insert(node(i,i,0,vis[i]));
	}
	for(int i=1;i<=t;i++){
		int op;
		scanf("%d",&op);
		if(op==1){
			int id,l,r,p;
			scanf("%d%d%d%d",&id,&l,&r,&p);
			solve(id,l,r,p);
		}else{
			int id;
			scanf("%d",&id);
			printf("%lld\n",w[id]);
		}
	}
	for(int i=1;i<=n;i++){
		if(find(i)==i)printf("%d\n",i);
	}
	return 0;
}