bzoj2588

题目:http://www.lydsy.com/JudgeOnline/problem.php?id=2588

就是静态区间第K大的变形。

每个节点为一棵线段树,表示到根的路径中,以权值为下标的线段树。

每个节点建树的时候,以父亲为历史版本。

对于询问点x和点y的时候,就是求ask(x)+ask(y)-ask(lca)-ask(fa[lca]),然后用类似于静态区间第K的方法决定是去左子树还是右子树。

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<utility>
#include<set>
#include<bitset>
#include<vector>
#include<functional>
#include<deque>
#include<cctype>
#include<climits>
#include<complex>
//#include<bits/stdc++.h>适用于CF,UOJ,但不适用于poj

using namespace std;

typedef long long LL;
typedef double DB;
typedef pair<int,int> PII;
typedef complex<DB> CP;

#define mmst(a,v) memset(a,v,sizeof(a))
#define mmcy(a,b) memcpy(a,b,sizeof(a))
#define re(i,a,b)  for(i=a;i<=b;i++)
#define red(i,a,b) for(i=a;i>=b;i--)
#define fi first
#define se second
#define m_p(a,b) make_pair(a,b)
#define SF scanf
#define PF printf
#define two(k) (1<<(k))

template<class T>inline T sqr(T x){return x*x;}
template<class T>inline void upmin(T &t,T tmp){if(t>tmp)t=tmp;}
template<class T>inline void upmax(T &t,T tmp){if(t<tmp)t=tmp;}

const DB EPS=1e-9;
inline int sgn(DB x){if(abs(x)<EPS)return 0;return(x>0)?1:-1;}
const DB Pi=acos(-1.0);

inline int gint()
  {
        int res=0;bool neg=0;char z;
        for(z=getchar();z!=EOF && z!=‘-‘ && !isdigit(z);z=getchar());
        if(z==EOF)return 0;
        if(z==‘-‘){neg=1;z=getchar();}
        for(;z!=EOF && isdigit(z);res=res*10+z-‘0‘,z=getchar());
        return (neg)?-res:res;
    }
inline LL gll()
  {
      LL res=0;bool neg=0;char z;
        for(z=getchar();z!=EOF && z!=‘-‘ && !isdigit(z);z=getchar());
        if(z==EOF)return 0;
        if(z==‘-‘){neg=1;z=getchar();}
        for(;z!=EOF && isdigit(z);res=res*10+z-‘0‘,z=getchar());
        return (neg)?-res:res;
    }

const int maxN=100000;

int N,M;
int a[maxN+100];
int bak[maxN+100],cnt;
int first[maxN+100],now;
struct Tedge{int v,next;}edge[2*maxN+100];

inline void addedge(int u,int v){now++;edge[now].v=v;edge[now].next=first[u];first[u]=now;}

int head,tail,que[maxN+100];
int fa[maxN+100],dep[maxN+100];
int jump[35][maxN+100];

struct Tnode{int val,son[2];}sn[maxN*70+100000];int idx;

int root[maxN+100];

inline void update(int past,int p,int l,int r,int x)
  {
      if(x<=l && r<=x){sn[p].val=sn[past].val+1;return;}
      int mid=(l+r)/2,f=(x<=mid);
      sn[p].son[f]=sn[past].son[f];
      sn[p].son[f^1]=++idx;
      if(x<=mid) update(sn[past].son[0],sn[p].son[0],l,mid,x); else update(sn[past].son[1],sn[p].son[1],mid+1,r,x);
      sn[p].val=sn[sn[p].son[0]].val+sn[sn[p].son[1]].val;
  }
inline int ask(int r1,int r2,int r3,int r4,int l,int r,int k)
  {
      if(l==r)return bak[l];
      int mid=(l+r)/2;
      int ge=sn[sn[r1].son[0]].val+sn[sn[r2].son[0]].val-sn[sn[r3].son[0]].val-sn[sn[r4].son[0]].val;
      if(ge<k)
        return ask(sn[r1].son[1],sn[r2].son[1],sn[r3].son[1],sn[r4].son[1],mid+1,r,k-ge);
      else
        return ask(sn[r1].son[0],sn[r2].son[0],sn[r3].son[0],sn[r4].son[0],l,mid,k);
  }

inline void BFS()
  {
      int i,j;
      mmst(fa,-1);
      fa[que[head=tail=0]=1]=0;
      dep[1]=1;
      while(head<=tail)
        {
            int u=que[head++],v;
            for(i=first[u],v=edge[i].v;i!=-1;i=edge[i].next,v=edge[i].v)if(fa[v]==-1){fa[que[++tail]=v]=u;dep[v]=dep[u]+1;}
        }
      re(i,0,N)root[i]=++idx;
      re(i,0,tail)
          {
              int u=que[i];
                update(root[fa[u]],root[u],1,cnt,a[u]);
            }
      re(j,0,30)jump[j][1]=1;
      re(i,1,tail)
        {
            int u=que[i];
            jump[0][u]=fa[u];
            re(j,1,30)jump[j][u]=jump[j-1][jump[j-1][u]];
        }
  }

inline void swim(int &x,int H){int i;for(i=0;H!=0;H/=2,i++)if(H&1)x=jump[i][x];}
inline int ask_lca(int x,int y)
  {
      if(dep[x]<dep[y])swap(x,y);
      swim(x,dep[x]-dep[y]);
      if(x==y)return x;
      int i;
      red(i,30,0)if(jump[i][x]!=jump[i][y]){x=jump[i][x];y=jump[i][y];}
      return jump[0][x];
  }

int main()
  {
      freopen("bzoj2588.in","r",stdin);
      freopen("bzoj2588.out","w",stdout);
      int i;
      N=gint();M=gint();
      re(i,1,N)a[i]=gint();
      re(i,1,N)bak[i]=a[i];
      sort(bak+1,bak+N+1);
      cnt=unique(bak+1,bak+N+1)-bak-1;
      re(i,1,N)a[i]=lower_bound(bak+1,bak+cnt+1,a[i])-bak;
      mmst(first,-1);now=-1;
      re(i,1,N-1){int x=gint(),y=gint();addedge(x,y);addedge(y,x);}
      BFS();
      int lastans=0;
      while(M--)
        {
            int x=gint()^lastans,y=gint(),lca=ask_lca(x,y),k=gint();
            lastans=ask(root[x],root[y],root[lca],root[fa[lca]],1,cnt,k);
            printf("%d",lastans);if(M!=0)printf("\n");
        }
      return 0;
  }

时间: 2024-10-12 22:30:31

bzoj2588的相关文章

BZOJ2588 Spoj 10628. Count on a tree

本文版权归ljh2000和博客园共有,欢迎转载,但须保留此声明,并给出原文链接,谢谢合作. 本文作者:ljh2000 作者博客:http://www.cnblogs.com/ljh2000-jump/转载请注明出处,侵权必究,保留最终解释权! 题目链接:BZOJ2588 SPOJ10628 正解:主席树上树 解题报告: 考虑主席树上树,每次build都是儿子在父亲的基础上build,然后查询的话就是对于x到y的路径,令$Tx$为$x$对应的线段树,那就是$T_x+T_y-T_{lca}-T_{f

【BZOJ2588】Spoj 10628. Count on a tree 主席树+LCA

[BZOJ2588]Spoj 10628. Count on a tree Description 给定一棵N个节点的树,每个点有一个权值,对于M个询问(u,v,k),你需要回答u xor lastans和v这两个节点间第K小的点权.其中lastans是上一个询问的答案,初始为0,即第一个询问的u是明文. Input 第一行两个整数N,M. 第二行有N个整数,其中第i个整数表示点i的权值. 后面N-1行每行两个整数(x,y),表示点x到点y有一条边. 最后M行每行两个整数(u,v,k),表示一组

bzoj2588 -- 树链剖分+主席树

先将权值离散. 显然可以对于每个结点建一棵权值线段树存这个点到根结点的路径上的点权,询问时在线段树上二分,但这样时间是O(n2log2n)的. 然后想到用主席树优化,时间复杂度O(n*log2n). 代码: 1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<vector> 6 using namespace st

【bzoj2588】Count on a tree

Portal -->bzoj2588 Solution 不行我一定要来挂这道题qwq很气愤qwq(其实还不是因为自己蠢..) 额首先说一下正解 如果这个问题放在序列上面的话..直接离散化一下然后一个可持久化权值线段树就好了 然后放在树上的话,我们可以考虑处理树上点对问题的一个很常见的套路: \[ info(x\rightarrow y)=info(root\rightarrow x)+info(root\rightarrow y)-info(root\rightarrow lca)*2 \] 那

【BZOJ-2588】Count on a tree 主席树 + 倍增

2588: Spoj 10628. Count on a tree Time Limit: 12 Sec  Memory Limit: 128 MBSubmit: 3749  Solved: 873[Submit][Status][Discuss] Description 给定一棵N个节点的树,每个点有一个权值,对于M个询问(u,v,k),你需要回答u xor lastans和v这两个节点间第K小的点权.其中lastans是上一个询问的答案,初始为0,即第一个询问的u是明文. Input 第一行

【BZOJ2588】【Spoj 10628.】 Count on a tree 可持久化线段树+lca

链接: #include <stdio.h> int main() { puts("转载请注明出处[vmurder]谢谢"); puts("网址:blog.csdn.net/vmurder/article/details/45048639"); } 题解: 对于每个树上节点存一个版本的可持久化线段树,为它到根节点上所有权值的权值线段树(需要离散化). 然后对于每次询问,这条链(a,b)的线段树就是:线段树a+线段树b?线段树lca?线段树falca 然后

【主席树】bzoj2588 Spoj 10628. Count on a tree

每个点的主席树的root是从其父转移来的.询问的时候用U+V-LCA-FA(LCA)即可. #include<cstdio> #include<algorithm> using namespace std; #define N 100001 int v[N<<1],first[N],next[N<<1],en,Ans; void AddEdge(int U,int V) { v[++en]=V; next[en]=first[U]; first[U]=en;

spoj COT/bzoj2588 Count on a tree

这题是学主席树的时候就想写的,,, 但是当时没写(懒) 现在来填坑 = =日常调半天lca(考虑以后背板) 主席树还是蛮好写的,但是代码出现重复,不太好,导致调试的时候心里没底(虽然事实证明主席树部分没出问题) 1 #include <cstdio> 2 #include <algorithm> 3 using namespace std; 4 struct lis{ int x,id;} li[100001]; 5 struct nod{ int num,ls,rs;} t[80

bzoj2588: Spoj 10628. Count on a tree(树上第k大)(主席树)

每个节点继承父节点的树,则答案为query(root[x]+root[y]-root[lca(x,y)]-root[fa[lca(x,y)]]) #include<iostream> #include<cstring> #include<cstdlib> #include<cstdio> #include<algorithm> using namespace std; const int maxn=100010; struct poi{int s