博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
[SDOI2013]森林(树上主席树+启发式合并+lca)
阅读量:4134 次
发布时间:2019-05-25

本文共 7003 字,大约阅读时间需要 23 分钟。

链接:https://ac.nowcoder.com/acm/problem/20577

来源:牛客网

题目描述

小Z有一片森林,含有N个节点,每个节点上都有一个非负整数作为权值。初始的时候,森林中有M条边。
小Z希望执行T个操作,操作有两类:

Q x y k查询点x到点y路径上所有的权值中,第k小的权值是多少。此操作保证点x和点y连通,同时这两个节点的路径上至少有k个点。

L x y在点x和点y之间连接一条边。保证完成此操作后,仍然是一片森林。
为了体现程序的在线性,我们把输入数据进行了加密。设lastans为程序上一次输出的结果,初始的时候lastans为0。

对于一个输入的操作Q x y k,其真实操作为Q x^lastans y^lastans k^lastans。

对于一个输入的操作L x y,其真实操作为L x^lastans ylastans。其中运算符表示异或,等价于pascal中的xor运算符。
请写一个程序來帮助小Z完成这些操作。
对于所有的数据,n,m,T<= 8∗10^4.

输入描述:

第一行包含一个正整数testcase,表示当前测试数据的测试点编号。保证1<=testcase<=20。
第二行包含三个整数N,M,T,分别表示节点数、初始边数、操作数。
第三行包含N个非负整数表示 N个节点上的权值。
接下来 M行,每行包含两个整数x和 y,表示初始的时候,点x和点y 之间有一条无向边。
接下来 T行,每行描述一个操作,格式为”Q x y k“或者”L x y “,其含义见题目描述部分。
输出描述:
对于每一个第一类操作,输出一个非负整数表示答案。
示例1
输入
复制
1
8 4 8
1 1 2 2 3 3 4 4
4 7
1 8
2 4
2 1
Q 8 7 3 Q 3 5 1
Q 10 0 0
L 5 4
L 3 2 L 0 7
Q 9 2 5 Q 6 1 6
输出
复制
2
2
1
4
2
备注:
在这里插入图片描述
很毒瘤的一道题目。
第一个输入的不是组数,而是样例的测试编号。。
对于求路径上第k小,就是主席树的板子。
但是对于合并两棵树,就需要点技巧的。类似于并查集一样的,就需要小的往大的上面合并。这样的话,耗时是最小的。总之这道题目很烦,不知道哪儿写错了,在洛谷上面只拿到了70分。。
代码如下:
洛谷70分

#include
#define ll long longusing namespace std;const int maxx=1e5+100;struct node{
int l; int r; int num;}p[maxx*600];struct edge{
int next,to;}e[maxx<<1];int head[maxx<<1],a[maxx],b[maxx],fa[maxx];int size[maxx],deep[maxx],dp[maxx][25],vis[maxx],root[maxx];int n,m,tot,ror,k,len;/*-----------事前准备----------*/inline void init(){
memset(head,-1,sizeof(head)); memset(dp,0,sizeof(dp)); memset(deep,0,sizeof(deep)); memset(size,0,sizeof(size)); memset(vis,0,sizeof(vis)); tot=ror=0;}inline void add(int u,int v){
e[tot].to=v,e[tot].next=head[u],head[u]=tot++;}inline int read(){
int x=0;char ch=' ';int f=1; while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar(); if(ch=='-')f=-1,ch=getchar(); while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar(); return x*f;}/*----------主席树-----------*/inline int build(int l,int r){
int cur=++ror; p[cur].num=0; if(l==r) return cur; int mid=l+r>>1; p[cur].l=build(l,mid); p[cur].r=build(mid+1,r); return cur;}inline int update(int rot,int l,int r,int pos){
int cur=++ror; p[cur]=p[rot]; p[cur].num++; if(l==r) return cur; int mid=l+r>>1; if(pos<=mid) p[cur].l=update(p[rot].l,l,mid,pos); else p[cur].r=update(p[rot].r,mid+1,r,pos); return cur;}inline int query(int lrot,int rrot,int frot,int ffrot,int l,int r,int k){
if(l==r) return l; int mid=l+r>>1; int ret=p[p[lrot].l].num+p[p[rrot].l].num-p[p[frot].l].num-p[p[ffrot].l].num; if(k<=ret) return query(p[lrot].l,p[rrot].l,p[frot].l,p[ffrot].l,l,mid,k); else return query(p[lrot].r,p[rrot].r,p[frot].r,p[ffrot].r,mid+1,r,k-ret);}/*------------dfs------------*/inline void dfs(int u,int f,int rt){
deep[u]=deep[f]+1; dp[u][0]=f; size[rt]++;//不断记录以rt为根节点的树的节点个数有多少 for(int i=1;i<=20;i++) {
if(dp[u][i-1]) dp[u][i]=dp[dp[u][i-1]][i-1]; else break; } vis[u]=1; root[u]=update(root[f],1,len,lower_bound(b+1,b+1+len,a[u])-b); for(int i=head[u];i!=-1;i=e[i].next) {
int to=e[i].to; if(to==f) continue; dfs(to,u,rt); }}/*----------并查集路径压缩----------*/inline int getf(int u)//找根节点,很巧妙的一个地方{
return u==fa[u]?fa[u]:(fa[u]=getf(fa[u]));}/*-----------lca-----------*/inline int get_lca(int x,int y){
if(deep[x]
=0;i--) {
if(dp[x][i]!=dp[y][i]) {
x=dp[x][i]; y=dp[y][i]; } } return dp[x][0];}int main(){
int t,x,y,d;char c; t=read(); //while(t--) //{
n=read(),m=read(),k=read(); init(); for(int i=1;i<=n;i++) {
a[i]=read(); vis[i]=0; fa[i]=i; b[i]=a[i]; } sort(b+1,b+1+n); len=unique(b+1,b+1+n)-b-1; for(int i=1;i<=m;i++) {
x=read();y=read(); add(x,y),add(y,x); } root[0]=build(1,len); for(int i=1;i<=n;i++) //先把所有的树跑一边,记录深度,父节点以及节点个数 {
if(!vis[i]) {
dfs(i,0,i); fa[i]=i; } } int ans=0; while(k--) {
getchar(); scanf("%c",&c); if(c=='Q') {
x=read(),y=read(),d=read(); x^=ans,y^=ans,d^=ans; int Lca=get_lca(x,y); ans=b[query(root[x],root[y],root[Lca],root[dp[Lca][0]],1,len,d)]; printf("%d\n",ans); } else {
x=read(),y=read(); x^=ans,y^=ans; add(x,y);add(y,x); int t1=getf(x); int t2=getf(y); if(size[t1]

满分代码:

#include
#include
#include
#include
#include
#include
#define ll long longusing namespace std;inline int read(){ int x=0;char ch=' ';int f=1; while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar(); if(ch=='-')f=-1,ch=getchar(); while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar(); return x*f;}struct edge{ int to,next;}e[320001];int T,n,m,q,tot;int a[80001];int fa[80001];int son[80001];int head[80001];inline void addedge(int x,int y){ e[++tot].to=y;e[tot].next=head[x];head[x]=tot;}struct Node{ int size,ls,rs;}t[80001*600];int cnt;int root[80001];void build(int &now,int l,int r){ now=++cnt; t[now].size=0; if(l==r)return; int mid=(l+r)>>1; build(t[now].ls,l,mid); build(t[now].rs,mid+1,r);}void insert(int &now,int pre,int l,int r,int x){ now=++cnt; t[now]=t[pre]; t[now].size++; if(l==r)return; int mid=(l+r)>>1; if(x<=mid)insert(t[now].ls,t[pre].ls,l,mid,x); else insert(t[now].rs,t[pre].rs,mid+1,r,x);}int b[80001];int size;int query(int x,int y,int pre1,int pre2,int l,int r,int k){ if(l==r)return b[l]; int lsize=t[t[x].ls].size+t[t[y].ls].size-t[t[pre1].ls].size-t[t[pre2].ls].size; int mid=(l+r)>>1; if(k<=lsize)return query(t[x].ls,t[y].ls,t[pre1].ls,t[pre2].ls,l,mid,k); else return query(t[x].rs,t[y].rs,t[pre1].rs,t[pre2].rs,mid+1,r,k-lsize);}inline int Hash(int x){ return lower_bound(b+1,b+size+1,x)-b;}int find(int x){ return fa[x]==x?x:fa[x]=find(fa[x]);}int st[80001][17];int dep[80001];int vis[80001];void dfs(int x,int father,int rt){ st[x][0]=father; for(int k=1;k<=16;k++){ st[x][k]=st[st[x][k-1]][k-1]; } son[rt]++; dep[x]=dep[father]+1; fa[x]=father; vis[x]=1; insert(root[x],root[father],1,size,Hash(a[x])); for(int i=head[x];i;i=e[i].next){ int u=e[i].to; if(u==father)continue; dfs(u,x,rt); }}inline int getlca(int x,int y){ if(x==y)return x; if(dep[x]>dep[y])swap(x,y); for(int k=16;k>=0;k--){ if(dep[st[y][k]]>=dep[x]){ y=st[y][k]; } } if(x==y)return x; for(int k=16;k>=0;k--){ if(st[x][k]!=st[y][k]){ x=st[x][k]; y=st[y][k]; } } return st[x][0];}int main(){ T=read(); T=1; while(T--){ memset(head,0,sizeof(head)); memset(dep,0,sizeof(dep)); memset(vis,0,sizeof(vis)); memset(st,0,sizeof(st)); memset(son,0,sizeof(son)); tot=0; cnt=0; n=read();m=read();q=read(); for(int i=1;i<=n;i++){ a[i]=read(); b[i]=a[i]; fa[i]=i; } sort(b+1,b+n+1); size=unique(b+1,b+n+1)-b-1; for(int i=1;i<=m;i++){ int x=read(),y=read(); addedge(x,y);addedge(y,x); } build(root[0],1,size); for(int i=1;i<=n;i++){ if(!vis[i]){ dfs(i,0,i); fa[i]=i; } } int lastans=0; for(int i=1;i<=q;i++){ char ch[3]; int x,y,k; scanf("%s",ch); x=read()^lastans; y=read()^lastans; if(ch[0]=='Q'){ k=read()^lastans; int lca=getlca(x,y); lastans=query(root[x],root[y],root[lca],root[st[lca][0]],1,size,k); printf("%d\n",lastans); } else{ addedge(x,y); addedge(y,x); int u=find(x); int v=find(y); if(son[u]

乞求路过的大佬看看哪儿不对,跪谢。

十一假期开启,codeforces之路起航
努力加油a啊,(o)/~

转载地址:http://kytvi.baihongyu.com/

你可能感兴趣的文章