感谢以下文章作者：
http://blog.csdn.net/kuribohg/article/details/41458639
http://vfleaking.blog.163.com/blog/static/174807634201311011201627/
http://blog.csdn.net/jiangyuze831/article/details/41476865
http://hzwer.com/5259.html
做了树上的莫队，感觉对这个算法的思想理解更深了
先分块，不论怎么分，要求同一块中的节点之间的距离不超过O(n0.5)
然后将图的DFS序搞出来
然后将询问（u,v)，以u所在的块为第一关键字，以v在dfs序中的位置为第二关键字排序。
然后弄个初始状态，然后就在图上按照询问的顺序“爬”（从(u,v)的状态转移到(u‘,v‘)，细节见vfleaking文章）。
至于复杂度，和序列型莫队的分析是一样的，我们的时间开销主要花在“爬”上，我们将爬的开销分成两部分来算：
第一部分：(u,v)中u改变造成的开销，如果在同一块中转移，我们最多需要走O(n0.5)步，要走O(n)次；如果在块间转移，我们最多走O(n)步，要走O(n0.5)次。总的O(n1.5)
第二部分：(u,v)中v改变造成的开销，同一块中的所有点总的开销是O(n)（同一块中的v是按dfs序排的序），有O(n0.5)块，所以是O(n1.5)；不同块间走O(n0.5)次，每次O(n)，总的也是O(n1.5)
所以总的是O（n1.5)。
这题没说m的范围，开成和n一样要RE，记得开成它的两倍。
1 /**************************************************************
2 Problem: 3757
3 User: idy002
4 Language: C++
5 Result: Accepted
6 Time:17840 ms
7 Memory:16640 kb
8 ****************************************************************/
9
10 #include <cstdio>
11 #include <cmath>
12 #include <vector>
13 #include <algorithm>
14 #define P(p) (1<<(p))
15 #define maxn 100010
16 #define maxp 15
17 using namespace std;
18
19
20 int n, m;
21 int root;
22 vector<int> g[maxn];
23 int stat[maxn], clr[maxn], cnt[maxn], clr_tot;
24 int anc[maxn][maxp+1], depth[maxn], dfn[maxn], dfs_clock;
25 int mccno[maxn], mcc_len, mcc_tot;
26 int ans[maxn];
27
28 struct Qu {
29 int u, v, id;
30 int a, b;
31 bool operator<( const Qu & c ) const {
32 return mccno[u]<mccno[c.u] || ( mccno[u]==mccno[c.u] && dfn[v]<dfn[c.v] );
33 }
34 };
35 Qu qry[maxn];
36
37 void dfs( int u, int fa, vector<int> &remain ) {
38 dfn[u] = ++dfs_clock;
39 anc[u][0] = fa;
40 for( int p=1; p<=maxp; p++ )
41 anc[u][p] = anc[anc[u][p-1]][p-1];
42 depth[u] = depth[fa]+1;
43 vector<int> cur;
44 for( int t=0; t<g[u].size(); t++ ) {
45 int v = g[u][t];
46 if( v==fa ) continue;
47 dfs(v,u,cur);
48 if( cur.size()>mcc_len ) {
49 mcc_tot++;
50 while( !cur.empty() ) {
51 mccno[ cur.back() ] = mcc_tot;
52 cur.pop_back();
53 }
54 }
55 }
56 while( !cur.empty() ) {
57 remain.push_back( cur.back() );
58 cur.pop_back();
59 }
60 remain.push_back(u);
61 }
62
63 int lca( int u, int v ) {
64 if( depth[u]<depth[v] ) swap(u,v);
65 int t = depth[u]-depth[v];
66 for( int p=maxp; p>=0 && t; p-- )
67 if( t&(P(p)) ) u = anc[u][p];
68 if( u==v ) return u;
69 for( int p=maxp; p>=0 && anc[u][0]!=anc[v][0]; p-- )
70 if( anc[u][p]!=anc[v][p] ) u = anc[u][p], v = anc[v][p];
71 return anc[u][0];
72 }
73
74 void inv_sig( int u ) {
75 int c = clr[u];
76 int d = stat[u] ? -1 : 1;
77 stat[u] ^= 1;
78 if( cnt[c]==0 ) clr_tot++;
79 cnt[c] += d;
80 if( cnt[c]==0 ) clr_tot--;
81 }
82 void inverse( int u, int v ) { // inverse T(u,v)
83 int ca = lca(u,v);
84 for( ; u!=ca; u=anc[u][0] )
85 inv_sig(u);
86 for( ; v!=ca; v=anc[v][0] )
87 inv_sig(v);
88 }
89 void calc( int q ) {
90 inverse( qry[q-1].u, qry[q].u );
91 inverse( qry[q-1].v, qry[q].v );
92 int ca = lca( qry[q].u, qry[q].v );
93
94 inv_sig( ca );
95 ans[qry[q].id] = clr_tot;
96 if( qry[q].a != qry[q].b && cnt[qry[q].a] && cnt[qry[q].b] )
97 ans[qry[q].id]--;
98 inv_sig( ca );
99 }
100
101 int main() {
102 scanf( "%d%d", &n, &m );
103 mcc_len = (int)sqrt(n+1);
104 for( int i=1; i<=n; i++ )
105 scanf( "%d", clr+i );
106 for( int i=1,u,v; i<=n; i++ ) {
107 scanf( "%d%d", &u, &v );
108 if( u==0 ) root=v;
109 if( v==0 ) root=u;
110 if( u&&v ) {
111 g[u].push_back( v );
112 g[v].push_back( u );
113 }
114 }
115 for( int i=1; i<=m; i++ ) {
116 scanf( "%d%d%d%d", &qry[i].u, &qry[i].v, &qry[i].a, &qry[i].b );
117 qry[i].id = i;
118 }
119
120 vector<int> remain;
121 dfs( root, root, remain );
122 while( remain.size() ) {
123 mccno[ remain.back() ] = mcc_tot;
124 remain.pop_back();
125 }
126
127 sort( qry+1, qry+1+m );
128
129 qry[0].u = qry[0].v = qry[1].u;
130 for( int i=1; i<=m; i++ )
131 calc(i);
132 for( int i=1; i<=m; i++ )
133 printf( "%d\n", ans[i] );
134 }
(代码巨慢，应该是分块时拖下来的）
时间： 2024-10-13 07:33:54