題目鏈接:https://cn.vjudge.net/problem/UVALive-5031
You are given an undirected graph with N vertexes and M edges. Every vertex in this graph has an
integer value assigned to it at the beginning. You’re also given a sequence of operations and you need
to process them as requested. Here’s a list of the possible operations that you might encounter:
1. Deletes an edge from the graph.
The format is [D X], where X is an integer from 1 to M, indicating the ID of the edge that you
should delete. It is guaranteed that no edge will be deleted more than once.
2. Queries the weight of the vertex with K-th maximum value among all vertexes currently
connected with vertex X (including X itself).
The format is [Q X K], where X is an integer from 1 to N, indicating the id of the vertex, and
you may assume that K will always fit into a 32-bit signed integer. In case K is illegal, the value
for that query will be considered as undefined, and you should return 0 as the answer to that
query.
3. Changes the weight of a vertex.
The format is [C X V ], where X is an integer from 1 to N, and V is an integer within the range
[−106
, 106
].
The operations end with one single character, ‘E’, which indicates that the current case has ended.
For simplicity, you only need to output one real number — the average answer of all queries.
Input
There are multiple test cases in the input file. Each case starts with two integers N and M (1 ≤ N ≤
2 ∗ 104
, 0 ≤ M ≤ 6 ∗ 104
), the number of vertexes in the graph. The next N lines describes the initial
weight of each vertex (−106 ≤ weight[i] ≤ 106
). The next part of each test case describes the edges
in the graph at the beginning. Vertexes are numbered from 1 to N. The last part of each test case
describes the operations to be performed on the graph. It is guaranteed that the number of query
operations [Q X K] in each case will be in the range [1, 2 ∗ 105
], and there will be no more than 2 ∗ 105
operations that change the values of the vertexes [C X V ].
There will be a blank line between two successive cases. A case with N = 0, M = 0 indicates the
end of the input file and this case should not be processed by your program.
Output
For each test case, output one real number — the average answer of all queries, in the format as
indicated in the sample output. Please note that the result is rounded to six decimal places.
Explanation for samples:
For the first sample:
D 3 – deletes the 3rd edge in the graph (the remaining edges are (1, 2) and (2, 3))
Q 1 2 – finds the vertex with the second largest value among all vertexes connected with 1. The
answer is 20.
Q 2 1 – finds the vertex with the largest value among all vertexes connected with 2. The answer is
30.
D 2 – deletes the 2nd edge in the graph (the only edge left after this operation is (1, 2))
Q 3 2 – finds the vertex with the second largest value among all vertexes connected with 3. The
answer is 0 (Undefined).
C 1 50 – changes the value of vertex 1 to 50.
Q 1 1 – finds the vertex with the largest value among all vertex connected with 1. The answer is 50.
E – This is the end of the current test case. Four queries have been evaluated, and the answer to
this case is (20 + 30 + 0 + 50) / 4 = 25.000.
For the second sample, caution about the vertex with same weight:
Q 1 1 – the answer is 20
Q 1 2 – the answer is 20
Q 1 3 – the answer is 10
Sample Input
3 3
10
20
30
1 2
2 3
1 3
D 3
Q 1 2
Q 2 1
D 2
Q 3 2
C 1 50
Q 1 1
E
3 3
10
20
20
1 2
2 3
1 3
Q 1 1
Q 1 2
Q 1 3
E
0 0
Sample Output
Case 1: 25.000000
Case 2: 16.666667
【中文題意】
有一張n個節點m條邊的無向圖,每個結點都有一個整數權值。你的任務是執行一系列操作。操作有如下三種:
1.D X(1<=X <=m) 刪除ID爲X的邊,輸入保證每條邊至多被刪除1次。
2.Q X k(1<=X <=n),計算與節點X連通的結點中(包括X本身),第k大的權值,如果不存在,返回0 。
3.C X V (1<=x <=n)把結點X的權值改爲V
操作序列結束的標誌位單個字母E,節點編號爲1~n,邊編號爲1~m。
輸入格式
輸入第一行爲兩個整數n和m(1<=n<=20000,0<=m<=60000),以下n行每行有一個絕對值不超過10^6 的整數,即各結點的初始權值,以下m行每行有兩個整數,即一條邊的兩個端點。接下來是每條指令,以單個字母E結尾。保證Q和C的指令均不超過200000條,輸入結束標誌位n=m=0。
輸出格式
對於每組數據,輸出所有Q指令的計算結果的平均值,精確到小數點後6位。
【分析】
本題只需要設計離線算法,因此可以把操作的順序反過來處理,先讀入所有操作,執行所有的D操作得到最終的圖,然後按照逆序把這些邊逐步插入,並在恰當的時機執行Q和C操作。用一棵名次樹維護一個聯通分量中的點權,則C操作對應名次樹中一次修改操作(可以用一次刪除加一次插入來實現),Q操作對應kth操作,而執行D操作時,如果兩個端點已經是一個同一連通分量則無影響,否則將兩個端點對應的名次樹合併。
【AC代碼】
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
using namespace std;
struct Node
{
Node *ch[2];//左右子樹
int r;//隨機優先級
int v;//該點的值
int s;//該點的節點總數
Node(int v):v(v)
{
ch[0]=ch[1]=NULL;//左右子樹初始化爲空
r=rand();//優先級隨機
s=1;//只有本身一個結點
}
bool operator <(const Node &rhs)const
{
return r<rhs.r;
}
int cmp(int x)const
{
if(x==v)return -1;
return x<v?0:1;//0代表左邊,1代表右邊
}
void maintain()//更新結點域
{
s=1;
if(ch[0]!=NULL)s+=ch[0]->s;
if(ch[1]!=NULL)s+=ch[1]->s;
}
};
void rotate(Node* &o,int d)//對o結點進行旋轉,0代表左旋,1代表右旋,旋轉後更新結點域
{
Node *k=o->ch[d^1];
o->ch[d^1]=k->ch[d];
k->ch[d]=o;
o->maintain();
k->maintain();
o=k;
}
void insert(Node* &o,int x)//插入一個值爲x的結點
{
if(o==NULL)o=new Node(x);
else
{
int d=(x<o->v?0:1);//x< o->v 時d=0該節點被插入左邊,否則d=1插入右邊
insert(o->ch[d],x);
if(o->ch[d]>o)rotate(o,d^1);//如果被插入後優先級比父節點大,要發生旋轉
}
o->maintain();
}
void remove(Node* &o,int x)//刪除一個值爲x的節點
{
int d=o->cmp(x);
if(d==-1)//d==1時代表代表需刪除的節點爲當前節點
{
Node* u=o;
if(o->ch[0]!=NULL&&o->ch[1]!=NULL)//左右孩子非空
{
int d2=(o->ch[0] > o->ch[1] ?1:0);//左孩子與右孩子誰的優先級比較高
rotate(o,d2);//誰的高向另外的方向旋轉
remove(o->ch[d2],x);//旋轉後刪除x節點
}
else
{
if(o->ch[0]==NULL)o=o->ch[1];//左孩子爲空,o指向右孩子
else o=o->ch[0];//否則指向左孩子
delete u;
}
}
else
remove(o->ch[d],x);
if(o!=NULL)o->maintain();//刪除後o有孩子的話更新結點域
}
const int maxc = 500000 + 10;
struct Command
{
char type;
int x,p;
}commands[maxc];//記錄要執行的命令
const int maxn = 20000 + 10;//點數最大
const int maxm = 60000 + 10;//邊數最大
int n,m,weight[maxn],from[maxm],to[maxm],removed[maxm];
//利用並查集維護聯通關係
int pa[maxn];
int findset(int x)
{
return pa[x]!=x ? pa[x]=findset(pa[x]) : x;
}
Node *root[maxn];//Treap
int kth(Node* o,int k)//查找以o爲根節點的子樹中的第k大
{
if(o==NULL || k <= 0 ||k > o->s)return 0;//不合法情況
int s = (o->ch[1] == NULL ? 0 : o->ch[1]->s);
if(k==s+1)return o->v;//正好第k大爲當前節點
else if(k<=s)
{
return kth(o->ch[1],k);//第k大在右子樹上
}
else return kth(o->ch[0],k-s-1);//第k大在左子樹上
}
void mergeto(Node* &src,Node* &dest)//合併兩棵子樹
{
if(src->ch[0] !=NULL)mergeto(src->ch[0],dest);
if(src->ch[1] !=NULL)mergeto(src->ch[1],dest);
insert(dest,src->v);
delete src;
src =NULL;
}
void removetree(Node* &x)//刪除根節點爲x的子樹
{
if(x->ch[0] != NULL)removetree(x->ch[0]);
if(x->ch[1] != NULL)removetree(x->ch[1]);
delete x;
x = NULL;
}
void add_edge(int x)//增加第x條邊
{
int u=findset(from[x]),v=findset(to[x]);
if(u!=v)
{
if(root[u]->s < root[v]->s)
{
pa[u]=v;
mergeto(root[u],root[v]);
}
else
{
pa[v]=u;
mergeto(root[v],root[u]);
}
}
}
int query_cnt;
long long query_tot;
void query(int x,int k)//查詢與x聯通的所有結點中的第k大
{
query_cnt++;
query_tot+=kth(root[findset(x)],k);
}
void change_weight(int x,int v)//把結點x的值改爲v 操作爲:先刪除,後添加
{
int u=findset(x);
remove(root[u],weight[x]);
insert(root[u],v);
weight[x]=v;
}
int main()
{
int kase=0;
while(scanf("%d%d",&n,&m)==2 &&n)
{
for(int i=1;i<=n;i++)
{
scanf("%d",&weight[i]);
}
for(int i=1;i<=m;i++)
{
scanf("%d%d",&from[i],&to[i]);
}
memset(removed,0,sizeof(removed));
int c=0;
for(;;)
{
char type;
int x,p=0,v=0;
scanf(" %c",&type);
if(type=='E')
{
break;
}
scanf("%d",&x);
if(type=='D')removed[x]=1;
if(type=='Q')scanf("%d",&p);
if(type=='C')
{
scanf("%d",&v);
p = weight[x];
weight[x] = v;
}
commands[c++] = (Command){type,x,p};
}
//最終的圖
for(int i=1;i<=n;i++)
{
pa[i]=i;
if(root[i]!=NULL)removetree(root[i]);
root [i]=new Node(weight[i]);
}
for(int i=1;i<=m;i++)
{
if(!removed[i])add_edge(i);
}
//反向操作
query_tot = query_cnt=0;
for(int i=c-1; i >= 0;i--)
{
if(commands[i].type=='D')
add_edge(commands[i].x);
if(commands[i].type=='Q')
query(commands[i].x,commands[i].p);
if(commands[i].type=='C')
change_weight(commands[i].x,commands[i].p);
}
printf("Case %d: %.6lf\n",++kase,query_tot/(double)query_cnt);
}
return 0;
}