主席树,一些奇奇怪怪的TLE
查看原帖
主席树,一些奇奇怪怪的TLE
890346
Autumn_0930楼主2024/7/28 14:20

同一份代码,昨天写的过了(跑得嘎嘎快),今天写的就T了一个点

这是T了的代码,有人帮忙看看吗:

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
struct node{
	int ls,rs,v;
}tree[N*30];
int n,m,top=0,a[N],root[N];
int add(int rt){
	top++;
	tree[top]=tree[rt];
	return top;
}
int build(int rt,int l,int r){
	rt=++top;
	if(l==r){
		tree[rt].v=a[l];
		return top;
	}
	int mid=l+r>>1;
	tree[rt].ls=build(tree[rt].ls,l,mid);
	tree[rt].rs=build(tree[rt].rs,mid+1,r);
	return rt;
}
int upd(int rt,int l,int r,int x,int v){
	rt=add(rt);
	if(l==r){
		tree[rt].v=v;
		return rt;
	}
	int mid=l+r>>1;
	if(x<=mid) tree[rt].ls=upd(tree[rt].ls,l,mid,x,v);
	else tree[rt].rs=upd(tree[rt].rs,mid+1,r,x,v);
	return rt;
}
int query(int rt,int l,int r,int x){
	if(l==r) return tree[rt].v;
	int mid=l+r>>1;
	if(x<=mid) return query(tree[rt].ls,l,mid,x);
	else return query(tree[rt].rs,mid+1,r,x);
}
int main(){
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	root[0]=build(1,1,n);
	for(int i=1;i<=m;i++){
		int v,op,loc,val;
		cin>>v>>op>>loc;
		if(op==1){
			scanf("%d",&val);
			root[i]=upd(root[v],1,n,loc,val);
		}else{
			root[i]=root[v];
			printf("%d\n",query(root[v],1,n,loc));
		}
	}
	return 0;
}
2024/7/28 14:20
加载中...