【並查集】BZOJ4551-[Tjoi2016&Heoi2016]樹

NOIP太可怕了((( -口-)

題目連接

【題目大意】

給定一顆有根樹（根爲1），有如下兩種操做：

1. 標記操做：對某個結點打上標記（在最開始，只有結點1有標記，其餘結點均無標記，並且對於某個結點，能夠打屢次標記。）

2. 詢問操做：詢問某個結點最近的一個打了標記的祖先（這個結點自己也算本身的祖先）

 

【思路】

正着作不行就反方向來。先離線處理全部操做，算出最終某個點被標記了幾回。跑一次dfs算出最終狀態時每一個點最近的打了標記的祖先u[i]。

從後往前從新看操做，若是是標記操做，那麼當前節點標記數-1，若是標記數爲0了，那麼u[i]=fa[i]最近的祖先（用並查集來處理以前的更新）。若是爲詢問，則輸出當前最近的祖先（一樣能夠用並查集來作）。

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int MAXN=100000+50;
 4 vector<int> E[MAXN];
 5 int mark[MAXN];
 6 int query[MAXN],ans[MAXN]; 
 7 int u[MAXN],fa[MAXN],n,q;
 8 char op[MAXN];
 9 
10 void dfs(int x,int anc,int father)
11 {
12     fa[x]=father;
13     if (mark[x]>0) u[x]=x;
14         else u[x]=anc;
15     for (int i=0;i<E[x].size();i++)
16     {
17         int to=E[x][i];
18         if (to==fa[x]) continue;
19         dfs(to,u[x],x);
20     }
21 }
22 
23 int union_set(int x)
24 {
25     int r=x;
26     while (u[r]!=r) r=u[r];
27     int now=x;
28     while (u[now]!=r)
29     {
30         int tmp=u[now];
31         u[now]=r;
32         now=tmp;
33     }
34     return u[x];
35 }
36 
37 void init()
38 {
39     scanf("%d%d",&n,&q);
40     for (int i=1;i<n;i++)
41     {
42         int u,v;
43         scanf("%d%d",&u,&v);
44         E[u].push_back(v);
45         E[v].push_back(u);
46     }
47     mark[1]=1;
48     for (int i=1;i<=q;i++)
49     {
50         char tmp[1];
51         scanf("%s %d",tmp,&query[i]);
52         if (tmp[0]=='C') mark[query[i]]++;
53         op[i]=tmp[0];
54     }
55     dfs(1,1,0);
56     for (int i=1;i<=n;i++) cout<<u[i]<<endl;
57 }
58 
59 void solve()
60 {
61     memset(ans,0,sizeof(ans));
62     for (int i=q;i>=1;i--)
63     {
64         int now=query[i];
65         if (op[i]=='C')
66         {
67             mark[now]--;
68             if (!mark[now]) now=union_set(fa[now]);
69         }
70         else ans[++ans[0]]=union_set(now);
71     }    
72     for (int i=ans[0];i>=1;i--) printf("%d\n",ans[i]);
73 } 
74 
75 int main()
76 {
77     init();
78     solve();
79     return 0;
80 }