网络管理Network

网络管理Network

时间限制: 5 Sec 内存限制: 256 MB
题目描述

M公司是一个非常庞大的跨国公司，在许多国家都设有它的下属分支机构或部门。为了让分布在世界各地的N个部门之间协同工作，公司搭建了一个连接整个公司的通信网络。该网络的结构由N个路由器和N-1条高速光缆组成。每个部门都有一个专属的路由器，部门局域网内的所有机器都联向这个路由器，然后再通过这个通信子网与其他部门进行通信联络。该网络结构保证网络中的任意两个路由器之间都存在一条直接或间接路径以进行通信。 高速光缆的数据传输速度非常快，以至于利用光缆传输的延迟时间可以忽略。但是由于路由器老化，在这些路由器上进行数据交换会带来很大的延迟。而两个路由器之间的通信延迟时间则与这两个路由器通信路径上所有路由器中最大的交换延迟时间有关。作为M公司网络部门的一名实习员工，现在要求你编写一个简单的程序来监视公司的网络状况。该程序能够随时更新网络状况的变化信息（路由器数据交换延迟时间的变化），并且根据询问给出两个路由器通信路径上延迟第k大的路由器的延迟时间。

【任务】 你的程序从输入文件中读入N个路由器和N-1条光缆的连接信息，每个路由器初始的数据交换延迟时间Ti，以及Q条询问（或状态改变）的信息。并依次处理这Q条询问信息，它们可能是： 1. 由于更新了设备，或者设备出现新的故障，使得某个路由器的数据交换延迟时间发生了变化。 2. 查询某两个路由器a和b之间的路径上延迟第k大的路由器的延迟时间。

输入

第一行为两个整数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，此时路径上只有一个路由器。

输出

对于每一个第二种询问（k>0），输出一行。包含一个整数为相应的延迟时间。如果路径上的路由器不足k个，则输出信息“invalid request!”（全部小写不包含引号，两个单词之间有一个空格）。

样例输入

5 5
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
样例输出

3
2
2
invalid request!

提示
100% 测试数据满足N<=8000,Q<=3000,任意一个路由器在任何时刻都满足延迟时间小于10^8。对于所有询问满足0<=K<=N 。
40% 测试数据满足所有询问中1<=K<=5 。即路由器的延迟时间不会发生变化。
10% 测试数据满足N,Q<=80000

来源
CTSC2008

题解

带修改树上区间第K大裸题。树状数组套主席树即可解决。

代码

#include<iostream> 
#include<cstdio> 
#include<cstdlib> 
#include<cstring> 
#include<string> 
#include<cmath> 
#include<algorithm>
#define N 80010 
#define Q 30010
using namespace std;
int n,q,tot,ans,cnt,Tnum,s[N],num[N*2];
int k,dfn,la[N],fa[N*2],st[N],end[N];
int l1,l2,s1[50],s2[50],w[N][18],dep[N];
struct point{int lc,rc,size;}t[N*120];
struct node{int a,b;}map[N*2];
struct wbs{int k,a,b;}p[Q];

void add(int a,int b)
{
  map[++k].a=a;map[k].b=b;fa[k]=la[a];la[a]=k;
  map[++k].a=b;map[k].b=a;fa[k]=la[b];la[b]=k;
}

void dfs(int x)
{
  st[x]=++dfn;
  for(int i=1;i<=17;i++)
    w[x][i]=w[w[x][i-1]][i-1]; 
  for(int a=la[x];a;a=fa[a])
    if(w[x][0]!=map[a].b)
    { 
      w[map[a].b][0]=x;
      dep[map[a].b]=dep[x]+1;
      dfs(map[a].b);
    } 
  end[x]=dfn;
}

class seg_tree
{
  public:
  void modify(int &x,int pre,int l,int r,int des,int val)
  {
    x=pre?pre:++cnt;
    t[x]=t[pre];t[x].size+=val;
    if(l==r)return;
    int mid=l+r>>1;
    if(des<=mid)modify(t[x].lc,t[pre].lc,l,mid,des,val);
    else modify(t[x].rc,t[pre].rc,mid+1,r,des,val);
  }
  int qry(int l,int r,int val)
  {
    int sum=0,mid=l+r>>1;
    if(l==r)return num[l];
    for(int i=1;i<=l1;i++)sum+=t[t[s1[i]].lc].size;
    for(int i=1;i<=l2;i++)sum-=t[t[s2[i]].lc].size;
    if(val>sum)
    {
      for(int i=1;i<=l1;i++)s1[i]=t[s1[i]].rc;
      for(int i=1;i<=l2;i++)s2[i]=t[s2[i]].rc;
      return qry(mid+1,r,val-sum);
    }
    else
    {
      for(int i=1;i<=l1;i++)s1[i]=t[s1[i]].lc;
      for(int i=1;i<=l2;i++)s2[i]=t[s2[i]].lc;
      return qry(l,mid,val);
    }
  }
}T2;

class bit
{
  public:
  int qry(int a,int b,int c,int d,int k)
  {
    l1=0;l2=0;
    for(;a;a-=a&-a)s1[++l1]=a;
    for(;b;b-=b&-b)s1[++l1]=b;
    for(;c;c-=c&-c)s2[++l2]=c;
    for(;d;d-=d&-d)s2[++l2]=d;
    return T2.qry(1,Tnum,k);
  }
  void modify(int x,int y,int val)
  {
    for(;x<=n;x+=x&-x)T2.modify(x,x,1,Tnum,y,val);
  }
}T1;

int lca(int a,int b)
{
  if(dep[a]<dep[b])swap(a,b);
  for(int i=17;i>=0;i--)
    if(dep[w[a][i]]>=dep[b])a=w[a][i];
  if(a==b)return a;
  for(int i=17;i>=0;i--)
    if(w[a][i]!=w[b][i])a=w[a][i],b=w[b][i];
  return w[a][0];
}

int main()
{
  int a,b;
  scanf("%d%d",&n,&q);tot=n;cnt=n;
  for(int i=1;i<=n;i++)scanf("%d",&s[i]),num[i]=s[i];
  for(int i=1;i<n;i++)scanf("%d%d",&a,&b),add(a,b);
  for(int i=1;i<=q;i++)
  {
    scanf("%d%d%d",&p[i].k,&p[i].a,&p[i].b);
    if(!p[i].k)num[++tot]=p[i].b;
  }
  dep[1]=1;dfs(1);
  sort(num+1,num+tot+1);
  Tnum=unique(num+1,num+tot+1)-num-1;
  for(int i=1;i<=n;i++)
  {
    s[i]=lower_bound(num+1,num+Tnum+1,s[i])-num;
    T1.modify(st[i],s[i],1);T1.modify(end[i]+1,s[i],-1); 
  } 
  for(int i=1;i<=q;i++)
  {
    if(p[i].k)
    {
      int a=p[i].a,b=p[i].b,k=p[i].k,x=lca(a,b);
      int sum=dep[a]+dep[b]-(dep[x]<<1)+1;
      if(k>sum)printf("invalid request!\n");
      else printf("%d\n",T1.qry(st[a],st[b],st[x],st[w[x][0]],sum-k+1));
    }
    else
    {
      int x=p[i].a,y=lower_bound(num+1,num+Tnum+1,p[i].b)-num;
      T1.modify(st[x],s[x],-1);T1.modify(end[x]+1,s[x],1);
      T1.modify(st[x],s[x]=y,1);T1.modify(end[x]+1,y,-1);
    }
  }
  return 0;
}