1. 程式人生 > >洛谷 3613 - 睡覺困難綜合征

洛谷 3613 - 睡覺困難綜合征

long 得到 father {} 睡覺 a10 from spa air

該題是[NOI2014]起床困難綜合征的樹上加修改版

先說說《起床困難綜合征》,由於不同位運算之間不滿足交換律,故必須按順序執行操作

考慮位運算套路 —— 拆位,對於未知的初始值,它的每一位也是未知的,所以可以用兩個變量 $a_0, a_1$ 來當作初始值當前位為 $0$ 或 $1$ 時最終可以得到的值,最後貪心即可

附上部分代碼:

 1 int a1 = 0, a2 = - 1;
 2 for (int i = 1; i <= N; i ++) {
 3     scanf ("%s", opt + 1);
 4     int p = getnum ();
5 if (opt[1] == A) 6 a1 &= p, a2 &= p; 7 else if (opt[1] == O) 8 a1 |= p, a2 |= p; 9 else if (opt[1] == X) 10 a1 ^= p, a2 ^= p; 11 } 12 int ps = 0, ans = 0; 13 for (int j = 30; j >= 1; j --) { 14 if (a1 & (1 << (j - 1))) 15
ans += (1 << (j - 1)); 16 else if ((a2 & (1 << (j - 1))) && ps + (1 << (j - 1)) <= M) 17 ans += (1 << (j - 1)), ps += (1 << (j - 1)); 18 }

那麽現在將操作擴展到樹上,使用 $LCT$ 來維護其代表的鏈的 $a_0, a_1$ ,通過找規律可以發現:

設新的 $a_0, a_1$ 分別為 $f_0, f_1$ ,舊的分別為 $x.a_0, x.a_1$ 及 $y.a_0, y.a_1$ (其中 $x, y$ 為從左往右),現在要合並 $x, y$

對於 $f_0$ 很好想到要將 $x.a_0 \ opt \ y.a_0$ 與 $x.a_0 \ opt \ y.a_1$ 合並即可得解,公式即為 $(x.a_0 \& y.a_1) | (~ x.a_0 \& y.a_0)$ ,其中, $(x.a_0 \& y.a_t)$ 為通解,第二個的取反即為消去 $x.a_0$ 中為 $1$ 的位對第二個操作的影響

$pushup$ 寫完就是 $LCT$ 常規操作了,求答案也是貪心即可

註意:由於有 $makeroot$ 操作,又求答案順序一定是從左往右,所以還需要維護一下從右往左的答案,用於在 $reverse$ 的時候交換

代碼

  1 #include <iostream>
  2 #include <cstdio>
  3 #include <cstring>
  4 
  5 using namespace std;
  6 
  7 typedef long long LL;
  8 typedef unsigned long long uLL;
  9 
 10 const int MAXN = 1e05 + 10;
 11 
 12 // {&, |, ^} -> {1, 2, 3}
 13 int opt[MAXN]= {0};
 14 LL value[MAXN]= {0};
 15 
 16 struct optionSt {
 17     LL a0, a1;
 18 
 19     optionSt () {}
 20     optionSt (LL fa0, LL fa1) :
 21         a0 (fa0), a1 (fa1) {}
 22 
 23 } ;
 24 optionSt merge (optionSt x, optionSt y) {
 25     LL f0 = ((x.a0 & y.a1) | (~ x.a0 & y.a0));
 26     LL f1 = ((x.a1 & y.a1) | (~ x.a1 & y.a0));
 27     return optionSt (f0, f1);
 28 }
 29 /*pair<LL, LL> merge (pair<LL, LL> p1, pair<LL, LL> p2) {
 30     LL a00 = p1.first, a10 = p1.second;
 31     LL a01 = p2.first, a11 = p2.second;
 32     LL a0 = ((a00 & a11) | (~ a00 & a01));
 33     LL a1 = ((a10 & a11) | (~ a10 & a01));
 34     return make_pair (a0, a1);
 35 }*/
 36 optionSt calc (int op, LL val) {
 37     switch (op) {
 38         case 1:
 39             return optionSt (0ll, val);
 40         case 2:
 41             return optionSt (val, - 1ll);
 42         case 3:
 43             return optionSt (val, ~ val);
 44     }
 45 }
 46 
 47 int father[MAXN]= {0};
 48 int son[MAXN][2]= {0};
 49 optionSt tval[MAXN], froml[MAXN], fromr[MAXN];
 50 int revtag[MAXN]= {0};
 51 
 52 int isroot (int p) {
 53     return son[father[p]][0] != p && son[father[p]][1] != p;
 54 }
 55 int sonbel (int p) {
 56     return son[father[p]][1] == p;
 57 }
 58 void reverse (int p) {
 59     if (! p)
 60         return ;
 61     swap (son[p][0], son[p][1]);
 62     swap (froml[p], fromr[p]); // 註意要更改左右順序
 63     revtag[p] ^= 1;
 64 }
 65 void pushup (int p) {
 66     froml[p] = fromr[p] = tval[p];
 67     if (son[p][0]) {
 68         froml[p] = merge (froml[son[p][0]], froml[p]);
 69         fromr[p] = merge (fromr[p], fromr[son[p][0]]);
 70     }
 71     if (son[p][1]) {
 72         froml[p] = merge (froml[p], froml[son[p][1]]);
 73         fromr[p] = merge (fromr[son[p][1]], fromr[p]);
 74     }
 75 }
 76 void pushdown (int p) {
 77     if (revtag[p]) {
 78         reverse (son[p][0]), reverse (son[p][1]);
 79         revtag[p] = 0;
 80     }
 81 }
 82 void rotate (int p) {
 83     int fa = father[p], anc = father[fa];
 84     int s = sonbel (p);
 85     son[fa][s] = son[p][s ^ 1];
 86     if (son[fa][s])
 87         father[son[fa][s]] = fa;
 88     if (! isroot (fa))
 89         son[anc][sonbel (fa)] = p;
 90     father[p] = anc;
 91     son[p][s ^ 1] = fa, father[fa] = p;
 92     pushup (fa), pushup (p);
 93 }
 94 int Stack[MAXN];
 95 int top = 0;
 96 void splay (int p) {
 97     top = 0, Stack[++ top] = p;
 98     for (int nd = p; ! isroot (nd); nd = father[nd])
 99         Stack[++ top] = father[nd];
100     while (top > 0)
101         pushdown (Stack[top]), top --;
102     for (int fa = father[p]; ! isroot (p); rotate (p), fa = father[p])
103         if (! isroot (fa))
104             sonbel (p) == sonbel (fa) ? rotate (fa) : rotate (p);
105     pushup (p);
106 }
107 void Access (int p) {
108     for (int tp = 0; p; tp = p, p = father[p])
109         splay (p), son[p][1] = tp, pushup (p);
110 }
111 void Makeroot (int p) {
112     Access (p), splay (p), reverse (p);
113 }
114 void Split (int x, int y) {
115     Makeroot (x);
116     Access (y), splay (y);
117 }
118 void link (int x, int y) {
119     Makeroot (x);
120     father[x] = y;
121 }
122 
123 int N, M, K;
124 
125 uLL Query (int x, int y, LL lim) {
126     Split (x, y);
127     uLL ans = 0, ps = 0;
128     LL a0 = froml[y].a0, a1 = froml[y].a1;
129     for (int j = K; j >= 1; j --) {
130         if (a0 & (1ll << (j - 1)))
131             ans += (1ll << (j - 1));
132         else if ((a1 & (1ll << (j - 1))) && ps + (1ll << (j - 1)) <= lim)
133             ans += (1ll << (j - 1)), ps += (1ll << (j - 1));
134     }
135     return ans;
136 }
137 
138 int getint () {
139     int num = 0;
140     char ch = getchar ();
141 
142     while (! isdigit (ch))
143         ch = getchar ();
144     while (isdigit (ch))
145         num = (num << 3) + (num << 1) + ch - 0, ch = getchar ();
146 
147     return num;
148 }
149 LL getLL () {
150     LL num = 0;
151     char ch = getchar ();
152 
153     while (! isdigit (ch))
154         ch = getchar ();
155     while (isdigit (ch))
156         num = (num << 3) + (num << 1) + ch - 0, ch = getchar ();
157 
158     return num;
159 }
160 uLL getuLL () {
161     uLL num = 0;
162     char ch = getchar ();
163 
164     while (! isdigit (ch))
165         ch = getchar ();
166     while (isdigit (ch))
167         num = (num << 3) + (num << 1) + ch - 0, ch = getchar ();
168 
169     return num;
170 }
171 void work (uLL x) {
172     if (x >= 10)
173         work (x / 10);
174     putchar (x % 10 + 0);
175 }
176 void println (uLL x) {
177     work (x), puts ("");
178 }
179 
180 int main () {
181     N = getint (), M = getint (), K = getint ();
182     for (int i = 1; i <= N; i ++) {
183         opt[i] = getint (), value[i] = getLL ();
184         tval[i] = calc (opt[i], value[i]);
185     }
186     for (int i = 1; i < N; i ++) {
187         int u = getint (), v = getint ();
188         link (u, v);
189     }
190     for (int Case = 1; Case <= M; Case ++) {
191         int op = getint ();
192         if (op == 1) {
193             int x = getint (), y = getint ();
194             uLL lim = getLL ();
195             println (Query (x, y, lim));
196         }
197         else if (op == 2) {
198             int p = getint (), nopt = getint ();
199             LL delta = getLL ();
200             Access (p), splay (p);
201             opt[p] = nopt, value[p] = delta;
202             tval[p] = calc (opt[p], value[p]);
203             pushup (p);
204         }
205     }
206 
207     return 0;
208 }
209 
210 /*
211 5 5 3
212 1 7
213 2 6
214 3 7
215 3 6
216 3 1
217 1 2
218 2 3
219 3 4
220 1 5
221 1 1 4 7
222 1 1 3 5
223 2 1 1 3
224 2 3 3 3
225 1 1 3 2
226 */
227 
228 /*
229 2 2 2
230 2 2
231 2 2
232 1 2
233 2 2 2 2
234 1 2 2 2
235 */

洛谷 3613 - 睡覺困難綜合征