码迷,mamicode.com
首页 > 其他好文 > 详细

T107741 【模板】权值线段树合并

时间:2019-11-07 13:41:45      阅读:89      评论:0      收藏:0      [点我收藏+]

标签:pac   image   ref   using   inf   合并   i++   ret   problem   

技术图片

技术图片

题目地址


#include<cstdio>
#include<iostream>
using namespace std;
const int MAXN=1e6;
struct Node{
	int ls,rs;
	int val,pos;
}tr[MAXN*20*4];
int nodeCnt=0;
void insert(int &p,int l,int r,int x,int val){
	if(!p)p=++nodeCnt;
	if(l==r){
		tr[p].val+=val;
		tr[p].pos=l;
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid)insert(tr[p].ls,l,mid,x,val);
	else if(x>mid)insert(tr[p].rs,mid+1,r,x,val);
	tr[p].val=max(tr[tr[p].ls].val,tr[tr[p].rs].val);
	tr[p].pos=tr[tr[p].ls].val>=tr[tr[p].rs].val?tr[tr[p].ls].pos:tr[tr[p].rs].pos;
}
int merge(int p,int q,int l,int r){
	if(!p)return q;
	if(!q)return p;
	if(l==r){
		tr[p].val+=tr[q].val;
		tr[p].pos=tr[p].val?l:0;
		return p;
	}
	int mid=(l+r)>>1;
	tr[p].ls=merge(tr[p].ls,tr[q].ls,l,mid);
	tr[p].rs=merge(tr[p].rs,tr[q].rs,mid+1,r);
	tr[p].val=max(tr[tr[p].ls].val,tr[tr[p].rs].val);
	tr[p].pos=tr[tr[p].ls].val>=tr[tr[p].rs].val?tr[tr[p].ls].pos:tr[tr[p].rs].pos;
	return p;
}
int roots[MAXN],rootCnt=0;
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		int opt;
		cin>>opt;
		if(opt==1){//新建线段树 
			roots[++rootCnt]=++nodeCnt;
		}else if(opt==2){//向某棵线段树中加入值 
			int nowRoot,x,num;
			scanf("%d%d%d",&nowRoot,&x,&num);
			insert(roots[nowRoot],1,MAXN,x,num);
		}else if(opt==3){//查询某棵线段树中出现最多的是哪个值 
			int nowRoot;
			scanf("%d",&nowRoot);
			cout<<tr[roots[nowRoot]].pos<<endl;
		}else if(opt==4){//合并两棵线段树 
			int root1,root2;
			scanf("%d%d",&root1,&root2);
			roots[++rootCnt]=merge(root1,root2,1,MAXN);
		}
	}
	return 0;
}

T107741 【模板】权值线段树合并

标签:pac   image   ref   using   inf   合并   i++   ret   problem   

原文地址:https://www.cnblogs.com/zbsy-wwx/p/11811236.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!