首页 > 代码库 > bzoj 3123: [Sdoi2013]森林
bzoj 3123: [Sdoi2013]森林
如果题号没记错的话,2588是一个树上的主席树查询。这个题就是多了个合并而已。每一次都把小的合并到大的上就好了(所谓启发式2333)
(主席树真是个好东西2333)
(上部分为本蒟蒻不知道为什么RE到死都RE的代码,,,挖坑)
(个人感觉主席树这种东西里离散不离散没什么区别的(常数而已),毕竟是log的,大个几次方就是多了个常数而已)
1 /*#include<bits/stdc++.h> 2 #define N 100005 3 #define M 40000005 4 #define LL long long 5 #define inf 0x3f3f3f3f 6 using namespace std; 7 inline int ra() 8 { 9 int x=0,f=1; char ch=getchar(); 10 while (ch<‘0‘ || ch>‘9‘) {if (ch==‘-‘) f=-1; ch=getchar();} 11 while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘; ch=getchar();} 12 return x*f; 13 } 14 int sz,n,m,Q,T,cnt,ans; 15 int ls[M],rs[M],sum[M],root[N],head[N]; 16 int size[N],belong[N],fa[N][20],vis[N],deep[N],a[N],hash[N],num[N],tot; 17 struct node{int to,next;}e[N<<2]; 18 void ins(int x, int y){e[++cnt].next=head[x]; e[cnt].to=y; head[x]=cnt;} 19 int find(int x) 20 { 21 int l=1,r=tot; 22 while (l<=r) 23 { 24 int mid=l+r>>1; 25 if (hash[mid]==x) return mid; 26 else if (hash[mid]<x) l=mid+1; 27 else r=mid-1; 28 } 29 } 30 void insert(int l, int r, int x, int &y, int v) 31 { 32 y=++sz; 33 sum[y]=sum[x]+1; 34 ls[y]=ls[x]; rs[y]=rs[x]; 35 if (l==r) return; 36 int mid=l+r>>1; 37 if (v<=mid) insert(l,mid,ls[x],ls[y],v); 38 else insert(mid+1,r,rs[x],rs[y],v); 39 } 40 void dfs(int x, int bl) 41 { 42 for (int i=1; i<=16; i++) 43 fa[x][i]=fa[fa[x][i-1]][i-1]; 44 insert(1,tot,root[fa[x][0]],root[x],find(a[x])); 45 belong[x]=bl; size[belong[x]]++; 46 for (int i=head[x];i;i=e[i].next) 47 { 48 if (e[i].to==fa[x][0]) continue; 49 deep[e[i].to]=deep[x]+1; 50 fa[e[i].to][0]=x; 51 dfs(e[i].to,bl); 52 } 53 } 54 int getlca(int x, int y) 55 { 56 if (deep[x]<deep[y]) swap(x,y); 57 int t=deep[x]-deep[y]; 58 for (int i=0; i <=16; i++) 59 if (t&(1<<i)) x=fa[x][i]; 60 for (int i=16; i>=0; i--) 61 if (fa[x][i]!=fa[y][i]) 62 x=fa[x][i],y=fa[y][i]; 63 if (x==y) return x; 64 return fa[x][0]; 65 } 66 void query(int x, int y, int k) 67 { 68 int a1=root[x],b=root[y],z=getlca(x,y),w=a[z]; 69 int c=root[z]; 70 int l=1,r=tot; 71 while (l!=r) 72 { 73 int mid=l+r>>1; 74 int tmp=sum[ls[a1]]+sum[ls[b]]-2*sum[ls[c]]+(w>=l && w<=mid); 75 if (tmp>=k) r=mid,a1=ls[a1],b=ls[b],c=ls[c]; 76 else k-=tmp,l=mid+1,a1=rs[a1],b=rs[b],c=rs[c]; 77 } 78 printf("%d\n",ans=hash[l]); 79 } 80 void merge(int x, int y) 81 { 82 if (size[belong[x]]<size[belong[y]]) swap(x,y); 83 fa[y][0]=x; deep[y]=deep[x]+1; ins(x,y); ins(y,x); 84 dfs(y,belong[x]); 85 } 86 int main() 87 { 88 T=ra(); int ti=0; 89 while (T--) 90 { 91 n=ra(); m=ra(); Q=ra(); 92 for (int i=1; i<=n; i++) num[i]=a[i]=ra(); 93 sort(num+1,num+n+1); 94 hash[++tot]=num[1]; 95 for (int i=2; i<=n; i++) 96 if (num[i]!=num[i-1]) 97 hash[++tot]=num[i]; 98 for (int i=1; i<=m; i++) 99 { 100 int x=ra(),y=ra(); 101 ins(x,y); ins(y,x); 102 } 103 for (int i=1; i<=n; i++) 104 if (!fa[i][0]) dfs(i,i); 105 for (int i=1; i<=Q; i++) 106 { 107 char s[3]; scanf("%s",s); 108 int x=ra()^ans,y=ra()^ans; 109 if (s[0]==‘Q‘) 110 { 111 int k=ra()^ans; 112 query(x,y,k); 113 } 114 else merge(x,y); 115 } 116 } 117 return 0; 118 }*/ 119 120 121 122 123 124 125 126 127 128 #include<bits/stdc++.h> 129 #define N 100005 130 #define M 10000005 131 #define LL long long 132 #define inf 0x3f3f3f3f 133 using namespace std; 134 inline int ra() 135 { 136 int x=0,f=1; char ch=getchar(); 137 while (ch<‘0‘ || ch>‘9‘) {if (ch==‘-‘) f=-1; ch=getchar();} 138 while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘; ch=getchar();} 139 return x*f; 140 } 141 int n,m,T,ans,cnt,tot,sz,testcase,query_sum; 142 int f[N],size[N],deep[N],a[N],num[N],hash[N],root[N],fa[N][17]; 143 int ls[M],rs[M],sum[M]; 144 int head[N],list[N<<1],next[N<<1]; 145 struct node{ 146 int to,next; 147 }e[N<<1]; 148 int Find(int x) 149 { 150 int l=1,r=tot; 151 while (l<=r) 152 { 153 int mid=l+r>>1; 154 if (hash[mid]==x) return mid; 155 else if (hash[mid]<x) l=mid+1; 156 else r=mid-1; 157 } 158 } 159 void insert(int x, int y) 160 { 161 e[++cnt].next=head[x]; e[cnt].to=y; head[x]=cnt; 162 } 163 int find(int i) 164 { 165 if (!f[i]) size[i]=1,f[i]=i; 166 if (f[i]==i) return i; 167 return f[i]=find(f[i]); 168 } 169 void merge(int x, int y) 170 { 171 int p=find(x),q=find(y); 172 f[p]=q; size[q]+=size[p]; 173 } 174 void update(int l, int r, int x, int &y, int v) 175 { 176 y=++sz; 177 ls[y]=ls[x]; rs[y]=rs[x]; sum[y]=sum[x]+1; 178 if (l==r) return; 179 int mid=l+r>>1; 180 if (v<=mid) update(l,mid,ls[x],ls[y],v); 181 else update(mid+1,r,rs[x],rs[y],v); 182 } 183 void dfs(int x) 184 { 185 for (int i=1; i<=16; i++) 186 fa[x][i]=fa[fa[x][i-1]][i-1]; 187 update(1,tot,root[fa[x][0]],root[x],Find(a[x])); 188 for (int i=head[x];i;i=e[i].next) 189 { 190 if (e[i].to==fa[x][0]) continue; 191 deep[e[i].to]=deep[x]+1; 192 fa[e[i].to][0]=x; 193 dfs(e[i].to); 194 } 195 } 196 int lca(int x, int y) 197 { 198 if (deep[x]<deep[y]) swap(x,y); 199 int t=deep[x]-deep[y]; 200 for (int i=0; (1<<i)<=t; i++) 201 if ((1<<i)&t) x=fa[x][i]; 202 for (int i=16; i>=0; i--) 203 if (fa[x][i]!=fa[y][i]) 204 x=fa[x][i],y=fa[y][i]; 205 return x==y?x:fa[x][0]; 206 } 207 int query(int a, int b, int c, int d, int k) 208 { 209 int l=1,r=tot; 210 while (l!=r) 211 { 212 int mid=l+r>>1; 213 int tmp=sum[ls[a]]+sum[ls[b]]-sum[ls[c]]-sum[ls[d]]; 214 if (tmp>=k) r=mid,a=ls[a],b=ls[b],c=ls[c],d=ls[d]; 215 else l=mid+1,k-=tmp,a=rs[a],b=rs[b],c=rs[c],d=rs[d]; 216 } 217 return l; 218 } 219 int main() 220 { 221 int textcast=ra(); 222 n=ra(); m=ra(); T=ra(); 223 for (int i=1; i<=n; i++) a[i]=num[i]=ra(); 224 sort(num+1,num+n+1); 225 hash[++tot]=num[1]; 226 for (int i=2; i<=n; i++) 227 if (num[i]!=num[i-1]) hash[++tot]=num[i]; 228 for (int i=1; i<=m; i++) 229 { 230 int x=ra(),y=ra(); 231 insert(x,y); insert(y,x); merge(x,y); 232 } 233 for (int i=1; i<=n; i++) 234 if (!fa[i][0]) dfs(i); 235 while (T--) 236 { 237 char s[3]; scanf("%s",s); 238 int a=ra()^ans,b=ra()^ans; 239 if (s[0]==‘Q‘) 240 { 241 int k=ra()^ans; 242 int c=lca(a,b); 243 int d=fa[c][0]; 244 printf("%d\n",ans=hash[query(root[a],root[b],root[c],root[d],k)]); 245 } 246 else 247 { 248 int p=find(a),q=find(b); 249 if (size[p]>size[q]) swap(a,b); 250 fa[a][0]=b; deep[a]=deep[b]+1; 251 dfs(a); 252 merge(a,b); 253 insert(a,b); insert(b,a); 254 } 255 } 256 return 0; 257 }
bzoj 3123: [Sdoi2013]森林
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。