「BZOJ1146」[CTSC2008] 网络管理Network
Description
M公司是一个非常庞大的跨国公司,在许多国家都设有它的下属分支机构或部门。为了让分布在世界各地的N个部门之间协同工作,公司搭建了一个连接整个公司的通信网络。该网络的结构由N个路由器和N-1条高速光缆组成。每个部门都有一个专属的路由器,部门局域网内的所有机器都联向这个路由器,然后再通过这个通信子网与其他部门进行通信联络。该网络结构保证网络中的任意两个路由器之间都存在一条直接或间接路径以进行通信。 高速光缆的数据传输速度非常快,以至于利用光缆传输的延迟时间可以忽略。但是由于路由器老化,在这些路由器上进行数据交换会带来很大的延迟。而两个路由器之间的通信延迟时间则与这两个路由器通信路径上所有路由器中最大的交换延迟时间有关。作为M公司网络部门的一名实习员工,现在要求你编写一个简单的程序来监视公司的网络状况。该程序能够随时更新网络状况的变化信息(路由器数据交换延迟时间的变化),并且根据询问给出两个路由器通信路径上延迟第k大的路由器的延迟时间。「任务」 你的程序从输入文件中读入N个路由器和N-1条光缆的连接信息,每个路由器初始的数据交换延迟时间Ti,以及Q条询问(或状态改变)的信息。并依次处理这Q条询问信息,它们可能是: 1. 由于更新了设备,或者设备出现新的故障,使得某个路由器的数据交换延迟时间发生了变化。 2. 查询某两个路由器a和b之间的路径上延迟第k大的路由器的延迟时间。
Input
第一行为两个整数N和Q,分别表示路由器总数和询问的总数。第二行有N个整数,第i个数表示编号为i的路由器初始的数据延迟时间Ti。紧接着N-1行,每行包含两个整数x和y。表示有一条光缆连接路由器x和路由器y。紧接着是Q行,每行三个整数k、a、b。如果k=0,则表示路由器a的状态发生了变化,它的数据交换延迟时间由Ta变为b。如果k>0,则表示询问a到b的路径上所经过的所有路由器(包括a和b)中延迟第k大的路由器的延迟时间。注意a可以等于b,此时路径上只有一个路由器。
Output
对于每一个第二种询问(k>0),输出一行。包含一个整数为相应的延迟时间。如果路径上的路由器不足k个,则输出信息“invalid request!”(全部小写不包含引号,两个单词之间有一个空格)。
Sample Input
5 1 2 3 4
3 1
2 1
4 3
5 3
2 4 5
0 1 2
2 2 3
2 1 4
3 3 5
Sample Output
2
2
invalid request!
HINT
题解
好像dfs序主席树啥的就能水过吧。。
不过好久没写了,所以还是写个nlogn^4的比较爽
二分+树链剖分+线段树套平衡树
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 |
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #define M 5000005 #define ll long long using namespace std; //==================================================== inline int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } //==================================================== int bin[20]; int n,q,cnt,tot,place,size,tmp; int T[100005],hash[100005]; int f[100005],a[100005],b[100005]; int fa[100005][17],son[100005],deep[100005],belong[100005],pl[100005]; int root[300005]; int w[M],v[M],s[M],rnd[M],ls[M],rs[M]; bool vis[100005]; //==================================================== struct data{int to,next;}e[200005];int head[100005]; void ins(int u,int v) { e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt; e[++cnt].to=u;e[cnt].next=head[v];head[v]=cnt; } //==================================================== 边表 int find(int x) { int l=1,r=tot; while(l<=r) { int mid=(l+r)>>1; if(hash[mid]<x)l=mid+1; else if(hash[mid]==x)return mid; else r=mid-1; } return l; } //==================================================== 二分查找 void dfs1(int x) { son[x]=1;vis[x]=1; for(int i=1;i<=16;i++) if(bin[i]<=deep[x])fa[x][i]=fa[fa[x][i-1]][i-1]; else break; for(int i=head[x];i;i=e[i].next) { if(vis[e[i].to])continue; deep[e[i].to]=deep[x]+1; fa[e[i].to][0]=x; dfs1(e[i].to); son[x]+=son[e[i].to]; } } void dfs2(int x,int chain) { place++;pl[x]=place;belong[x]=chain; int k=0; for(int i=head[x];i;i=e[i].next) if(son[e[i].to]>son[k]&&deep[e[i].to]>deep[x]) k=e[i].to; if(!k)return; dfs2(k,chain); for(int i=head[x];i;i=e[i].next) if(e[i].to!=k&&deep[e[i].to]>deep[x]) dfs2(e[i].to,e[i].to); } int lca(int x,int y) { if(deep[x]<deep[y])swap(x,y); int t=deep[x]-deep[y]; for(int i=0;i<=16;i++) if(bin[i]&t)x=fa[x][i]; for(int i=16;i>=0;i--) if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i]; if(x==y)return x; return fa[x][0]; } //================================================= 预处理 void update(int k){s[k]=s[ls[k]]+s[rs[k]]+w[k];} void rturn(int &k) {int t=ls[k];ls[k]=rs[t];rs[t]=k;update(k);update(t);k=t;} void lturn(int &k) {int t=rs[k];rs[k]=ls[t];ls[t]=k;update(k);update(t);k=t;} void insert(int &k,int num) { if(!k){k=++size;rnd[k]=rand();w[k]=s[k]=1;v[k]=num;return;} s[k]++; if(num==v[k]){w[k]++;return;} else if(num<v[k]){insert(ls[k],num);if(rnd[ls[k]]<rnd[k])rturn(k);} else {insert(rs[k],num);if(rnd[rs[k]]<rnd[k])lturn(k);} } void del(int &k,int num) { if(!k)return; if(num==v[k]) { if(w[k]>1){w[k]--;s[k]--;return;} if(ls[k]*rs[k]==0)k=ls[k]+rs[k]; else if(rnd[ls[k]]<rnd[rs[k]]){rturn(k);del(k,num);} else {lturn(k);del(k,num);} } else if(num<v[k]) {del(ls[k],num);s[k]--;} else {del(rs[k],num);s[k]--;} } void askrank(int k,int num) { if(!k)return; if(num==v[k]){tmp+=s[rs[k]];return;} else if(num<v[k]){tmp+=s[rs[k]]+w[k];askrank(ls[k],num);} else askrank(rs[k],num); } //==================================================== 平衡树 void change(int k,int l,int r,int pos,int x,int y) { del(root[k],x);insert(root[k],y); if(l==r)return; int mid=(l+r)>>1; if(pos<=mid)change(k<<1,l,mid,pos,x,y); else change(k<<1|1,mid+1,r,pos,x,y); } void ask(int k,int l,int r,int x,int y,int num) { if(x==l&&y==r){askrank(root[k],num);return;} int mid=(l+r)>>1; if(y<=mid)ask(k<<1,l,mid,x,y,num); else if(x>mid)ask(k<<1|1,mid+1,r,x,y,num); else {ask(k<<1,l,mid,x,mid,num);ask(k<<1|1,mid+1,r,mid+1,y,num);} } //==================================================== 线段树 void getrank(int x,int f,int num) { while(belong[x]!=belong[f]) { ask(1,1,n,pl[belong[x]],pl[x],num); x=fa[belong[x]][0]; } ask(1,1,n,pl[f],pl[x],num); } //====================================================树链剖分 void solve(int x,int y,int rank) { int t=lca(x,y),ans=-1; tmp=0;getrank(y,t,0);getrank(x,t,0); if(tmp-1<rank) {printf("invalid request!\n");return;} int l=1,r=tot; while(l<=r) { int mid=(l+r)>>1; tmp=0; getrank(x,t,mid); getrank(y,t,mid); if(T[t]>mid)tmp--; if(tmp<=rank-1){r=mid-1;ans=mid;} else l=mid+1; } printf("%d\n",hash[ans]); } //==================================================== int main() { bin[0]=1;for(int i=1;i<=16;i++)bin[i]=(bin[i-1]<<1); n=read();q=read(); for(int i=1;i<=n;i++) T[i]=read(),hash[++tot]=T[i]; for(int i=1;i<n;i++) { int u=read(),v=read(); ins(u,v); } dfs1(1);dfs2(1,1); for(int i=1;i<=q;i++) { f[i]=read(),a[i]=read(),b[i]=read(); if(!f[i])hash[++tot]=b[i]; } sort(hash+1,hash+tot+1); int top=1; for(int i=2;i<=tot;i++) if(hash[i]!=hash[i-1])hash[++top]=hash[i]; tot=top; for(int i=1;i<=n;i++)T[i]=find(T[i]); for(int i=1;i<=q;i++)if(!f[i])b[i]=find(b[i]); for(int i=1;i<=n;i++)change(1,1,n,pl[i],0,T[i]); for(int i=1;i<=q;i++) if(!f[i]) { change(1,1,n,pl[a[i]],T[a[i]],b[i]); T[a[i]]=b[i]; } else solve(a[i],b[i],f[i]); return 0; } |
为什么要hash?
权值离散化,减小枚举范围