spoj COT 可持久化数据结构 (LCA模版)

查询树链第K大 。                


每个版本的线段树维护的是 从这个节点到 根的 树链的版本, 由于树链第K大,在统计比X 小的数个数时 是可以 进行加减法运算的,所以  就可以用可持久化数据结构。

维护个数时 , sum = f(a) + f(b) - f(c) -f(d)    : c 为 a,b 的最近公共祖先, d 为 c 的父亲节点。这样就是 四个版本运算。

同时:二分可以直接在树上跑,判断 左半区域的和 是否大于K,大于K 说明第K大的值 还在 左区间, 相反在右区间里查 第K -sum 大的数。

复杂度 O(nlgn) 如果直接二分区间 复杂度是O(nlgnlgn)。

倍增 LCA 算法:

const int K = 18;
int d[maxn];
int p[maxn][K];
void dfs(int rt,int f){  
    d[rt]=d[f]+1;  
    p[rt][0]=f;  
    int pos = mp[num[rt]];
    root[rt] = update(pos,1,n,1,root[f]);
    for(int i=1;id[b]) swap(a,b);
    if(d[a]=0;i--){
            if(p[a][i]!= p[b][i]){
                a = p[a][i],b = p[b][i];
            }
        }
        a= p[a][0],b = p[b][0];
    }
    return a;
}

代码:

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define REP(i,n) for(int i=0;imp;
map::iterator it;
int idx[maxn];
int head[maxn];
struct Edge{
    int v;
    int next;
}edge[2*maxn];
int tot;
void init(){
    memset(head,-1,sizeof(head));
    CLEAR(d);
    CLEAR(p);
    tot = 0;
}
void add(int u,int v){
    tot ++;
    edge[tot].v= v;
    edge[tot].next = head[u];
    head[u] = tot;
}


struct Node{
	Node *l,*r;
	int sum;
}nodes[maxn*40];
Node *root[maxn];
Node *null;
int C;
void inits(){
	C= 0;
	null = &nodes[C++];
	root[0] = null;
	null->l = null->r = null;
	null->sum = 0;
}
Node *update(int pos,int left ,int right,int val,Node *root){
	 Node *rt = &nodes[C++];
	 rt->l = root->l;
	 rt->r = root->r;
	 rt->sum = root->sum;
	 if(left ==right){
	 	  rt->sum +=val;
	      return rt;
	 }
	 int mid =(left +right)/2;
	 if(pos<=mid){
	 	rt ->l =update(pos,left,mid,val,root->l);
	 }else{
	 	rt ->r = update(pos,mid+1,right,val,root->r);
	 }
	 rt->sum = rt->l->sum + rt->r->sum;
	 return rt;
}
int query(int k,int left ,int right,Node *rt,Node *rt2,Node *rt3,Node *rt4){
//	cout << left << " lr "<sum<<" "<< rt2->sum <<"   "<sum<<" "<sum<l->sum + rt2->l->sum - rt3->l->sum - rt4->l->sum;
	// cout << sum <<" sum k " << k << " "<=k){
	 	  return query(k,left,mid,rt->l,rt2->l,rt3->l,rt4->l);
	 }else{
	 	return query(k-sum,mid+1,right,rt->r,rt2->r,rt3->r,rt4->r);
	 }
}
int get(int a,int b,int c,int d,int  k){
	return query(k,1,n,root[a],root[b],root[c],root[d]);
}

void dfs(int rt,int f){  
    d[rt]=d[f]+1;  
    p[rt][0]=f;  
    int pos = mp[num[rt]];
    root[rt] = update(pos,1,n,1,root[f]);
    for(int i=1;id[b]) swap(a,b);
	if(d[a]=0;i--){
			if(p[a][i]!= p[b][i]){
				a = p[a][i],b = p[b][i];
			}
		}
		a= p[a][0],b = p[b][0];
	}
	return a;
}
void solve(){
    init();
	inits();
	for(int i =1;isecond = tot2;
        	//cout << tot2 << "  "<first<first; 
        }   
        
        solve();
    }
    return 0;
}


你可能感兴趣的:(数据结构)