參數 nod Go cin using log AR ios 格式

【問題描述】
小奇的花園有 n 個溫室,標號為 1 到 n,溫室以及以及溫室間的雙向道路形成一
棵樹。
每個溫室都種植著一種花,隨著季節的變換,溫室裏的花的種類也在不斷發生著
變化。
小奇想知道從溫室 x 走到溫室 y 的路徑中(包括兩個端點),第 t 種花出現的次數。
【輸入格式】
第一行為兩個整數 n,q,表示溫室的數目和操作的數目。
第二行有 n 個整數 T 1 ,T 2 ...T n 其中 T i 表示溫室 i 中的花的種類。接下來 n-1 行,每個兩個整數 x, y,表示溫室 x 和 y 之間有一條雙向道路。
接下來 q 行,表示 q 個操作,分別為以下兩種形式之一:
• C x t 表示在溫室 x 中的花的種類變為 t。

• Q x y t 表示詢問溫室 x 走到溫室 y 的路徑中(包括兩個端點),第 t 種花出現
的次數。
為了體現在線操作,輸入數據中的每個操作的參數都進行了加密。記最後一次詢
問的答案為 anslast(一開始設 anslast 為 0),下次讀入中的 x,y,t 均需要異或上
anslast 以得到真實值,在 C/C++中異或為ˆ運算符,在 pascal 中為 xor 運算符。
【輸出格式】
輸出 q 行,每行一個正整數表示該次詢問答案。
【樣例輸入】
5 8
10 20 30 40 50
1 2
1 3
3 4
3 5
Q 2 5 10
C 2 21
Q 3 4 21
C 6 22
Q 1 7 28
C 5 20
Q 2 5 20
Q 2 0 9
【樣例輸出】
1
2
0
3
1
【樣例解釋】
加密前的操作:
Q 2 5 10
C 3 20
Q 2 5 20
C 4 20
Q 3 5 30
C 5 20
Q 2 5 20Q 1 3 10
【數據範圍】
對於 30%的數據,有 n <= 1000, q <= 2000。
對於 50%的數據,有 n <= 10000, q <= 20000。
對於 100%的數據,有 n <= 100000, q <= 200000,0 <= T <= 2^31。

樹鏈剖分

對於每一種顏色建一棵線段樹，動態開點

顏色用map離散，最多300000種顏色

動態開店用內存池防止內存超限

  1 #include<iostream>
  2 #include<cstdio>
  3
 
 #include<cstring>
  4 #include<algorithm>
  5 #include<cmath>
  6 #include<map>
  7 #include<queue>
  8 using namespace std;
  9 struct Node
 10 {
 11   int next,to;
 12 }edge[200001];
 13 queue<int>Q;
 14 int num,head[100001],pos,tot,cnt,dep[100001],fa[100001],size[100001],son[100001];
 15 int top[100001],dfn[100001],root[300001],t[300001],ans,c[10000001],ch[10000001][2],n,q;
 16 map<int,int>mp;
 17 void add(int u,int v)
 18 {
 19   num++;
 20   edge[num].next=head[u];
 21   head[u]=num;
 22   edge[num].to=v;
 23 }
 24 void dfs1(int x,int pa)
 25 {int i;
 26   dep[x]=dep[pa]+1;
 27   fa[x]=pa;
 28   size[x]=1;
 29   for (i=head[x];i;i=edge[i].next)
 30     {
 31       int v=edge[i].to;
 32       if (v==pa) continue;
 33       dfs1(v,x);
 34       size[x]+=size[v]; 
 35       if (size[v]>size[son[x]]) son[x]=v;
 36     } 
 37 }
 38 void dfs2(int x,int pa,int tp)
 39 {int i;
 40   top[x]=tp;
 41   dfn[x]=++tot;
 42   if (son[x]) dfs2(son[x],x,tp);
 43   for (i=head[x];i;i=edge[i].next)
 44     {
 45       int v=edge[i].to;
 46       if (v==pa||son[x]==v) continue;
 47       dfs2(v,x,v);
 48     }
 49 }
 50 int query(int rt,int l,int r,int L,int R)
 51 {
 52   if (!rt) return 0;
 53   if (l>=L&&r<=R) return c[rt];
 54   int mid=(l+r)/2,as=0;
 55   if (L<=mid) as+=query(ch[rt][0],l,mid,L,R);
 56   if (R>mid) as+=query(ch[rt][1],mid+1,r,L,R);
 57   return as;
 58 }
 59 int ask(int x,int y,int c)
 60 {
 61   int as=0;
 62   while (top[x]!=top[y])
 63     {
 64       if (dep[top[x]]<dep[top[y]]) swap(x,y);
 65       as+=query(root[c],1,n,dfn[top[x]],dfn[x]);
 66       x=fa[top[x]];
 67     }
 68   if (dep[x]>dep[y]) swap(x,y);
 69   as+=query(root[c],1,n,dfn[x],dfn[y]);
 70   return as;
 71 }
 72 void update(int &rt,int l,int r,int x)
 73 {
 74   if (!rt)
 75     {
 76       if (Q.empty()==0) rt=Q.front(),Q.pop();
 77       else rt=++pos;
 78     }
 79   c[rt]++;
 80   if (l==r) return;
 81   int mid=(l+r)/2;
 82   if (x<=mid) update(ch[rt][0],l,mid,x);
 83   else update(ch[rt][1],mid+1,r,x);
 84 }
 85 void del(int &rt,int l,int r,int x)
 86 {
 87   c[rt]--;
 88   if (l==r&&c[rt]==0)
 89     {
 90       Q.push(rt);
 91       rt=0;
 92     }
 93   if (l==r) return;
 94   int mid=(l+r)/2;
 95   if (x<=mid) del(ch[rt][0],l,mid,x);
 96   else del(ch[rt][1],mid+1,r,x);
 97   if (c[rt]==0)
 98     {
 99       ch[rt][0]=ch[rt][1]=0;
100       Q.push(rt);
101       rt=0;
102     }
103 }
104 int main()
105 {int i,u,v,x,y,c;
106   char s[21];
107   cin>>n>>q;
108   for (i=1;i<=n;i++)
109     {
110       scanf("%d",&t[i]);
111       if (mp.count(t[i])==0) mp[t[i]]=++cnt;
112       t[i]=mp[t[i]];
113     }
114   for (i=1;i<=n-1;i++)
115     {
116       scanf("%d%d",&u,&v);
117       add(u,v);add(v,u);
118     }
119   dfs1(1,0);
120   dfs2(1,0,1);
121   for (i=1;i<=n;i++)
122     {
123       update(root[t[i]],1,n,dfn[i]);
124     }
125   ans=0;
126   while (q--)
127     {
128       scanf("%s",s);
129       if (s[0]==‘Q‘)
130     {
131       scanf("%d%d%d",&x,&y,&c);
132       x^=ans;y^=ans;c^=ans;
133       ans=0;
134       if (mp.count(c)==0)
135         {
136           printf("0\n");
137           continue;
138         }
139       c=mp[c];
140       ans=ask(x,y,c);
141       printf("%d\n",ans);
142     }
143       else
144     {
145       scanf("%d%d",&x,&c);
146       x^=ans;c^=ans;
147       del(root[t[x]],1,n,dfn[x]);
148       if (mp.count(c)==0) mp[c]=++cnt;
149       t[x]=mp[c];
150       update(root[t[x]],1,n,dfn[x]);
151     }
152     }
153 }

小奇的花園