【HDOJ】4605 Magic Ball Game

思路1：
树状数组+离线处理，对所有的w离散化处理，边dfs边使用树状数组更新左右w的情况。
思路2：
主席树，边bfs边建树。结点信息存储cnt，然后在线查询。
树状数组。

  1 /* 4605 */
  2 #include <iostream>
  3 #include <sstream>
  4 #include <string>
  5 #include <map>
  6 #include <queue>
  7 #include <set>
  8 #include <stack>
  9 #include <vector>
 10 #include <deque>
 11 #include <algorithm>
 12 #include <cstdio>
 13 #include <cmath>
 14 #include <ctime>
 15 #include <cstring>
 16 #include <climits>
 17 #include <cctype>
 18 #include <cassert>
 19 #include <functional>
 20 #include <iterator>
 21 #include <iomanip>
 22 using namespace std;
 23 //#pragma comment(linker,"/STACK:102400000,1024000")
 24 
 25 #define sti                set<int>
 26 #define stpii            set<pair<int, int> >
 27 #define mpii            map<int,int>
 28 #define vi                vector<int>
 29 #define pii                pair<int,int>
 30 #define vpii            vector<pair<int,int> >
 31 #define rep(i, a, n)     for (int i=a;i<n;++i)
 32 #define per(i, a, n)     for (int i=n-1;i>=a;--i)
 33 #define clr                clear
 34 #define pb                 push_back
 35 #define mp                 make_pair
 36 #define fir                first
 37 #define sec                second
 38 #define all(x)             (x).begin(),(x).end()
 39 #define SZ(x)             ((int)(x).size())
 40 #define lson            l, mid, rt<<1
 41 #define rson            mid+1, r, rt<<1|1
 42 
 43 typedef struct ques_t {
 44     int w, id;
 45     
 46     ques_t() {}
 47     ques_t(int w, int id):
 48         w(w), id(id) {}
 49         
 50 } ques_t;
 51 
 52 const int maxn = 1e5+5;
 53 const int maxm = maxn * 2;
 54 int a[maxm];
 55 int W[maxn];
 56 ques_t Q[maxn];
 57 vector<ques_t> vc[maxn];
 58 int ans[maxn][2];
 59 int son[maxn][2];
 60 int tot[maxm][2];
 61 int n, m;
 62 
 63 int lowest(int x) {
 64     return x & -x;
 65 }
 66 
 67 void add(int x, int delta, int id) {
 68     while (x <= m) {
 69         tot[x][id] += delta;
 70         x += lowest(x);
 71     }
 72 }
 73 
 74 int sum(int x, int id) {
 75     int ret = 0;
 76     
 77     while (x) {
 78         ret += tot[x][id];
 79         x -= lowest(x);
 80     }
 81     
 82     return ret;
 83 }
 84 
 85 void init() {
 86     memset(son, 0, sizeof(son));
 87     memset(tot, 0, sizeof(tot));
 88     rep(i, 1, n+1)
 89         vc[i].clr();
 90     m = 1;
 91 }
 92 
 93 void dfs(int u, int tot) {
 94     int v, sz = SZ(vc[u]);
 95     int llt, lgt, rlt, rgt;
 96     int eq;
 97     
 98     rep(i, 0, sz) {
 99         int w = vc[u][i].w;
100         int id = vc[u][i].id;
101         llt = sum(w-1, 0);
102         lgt = sum(m, 0) - sum(w, 0);
103         rlt = sum(w-1, 1);
104         rgt = sum(m, 1) - sum(w, 1);
105         eq = tot - llt - lgt - rlt - rgt;
106         if (eq) {
107             ans[id][0] = -1;
108         } else {
109             ans[id][0] = (lgt+rgt) + (llt+rlt)*3;
110             ans[id][1] = rlt;
111         }
112     }
113     if (son[u][0]) {
114         add(W[u], 1, 0);
115         dfs(son[u][0], tot+1);
116         add(W[u], -1, 0);
117         
118         add(W[u], 1, 1);
119         dfs(son[u][1], tot+1);
120         add(W[u], -1, 1);
121     }
122 }
123 
124 void solve() {
125     int q;
126     int x, w;
127     
128     scanf("%d", &q);
129     rep(i, 0, q) {
130         scanf("%d %d", &Q[i].id, &Q[i].w);
131         a[m++] = Q[i].w;
132     }
133     
134     sort(a+1, a+m);
135     m = unique(a+1, a+m) - (a+1);
136     
137     rep(i, 1, n+1) {
138         W[i] = lower_bound(a+1, a+1+m, W[i]) - a;
139     }
140     
141     rep(i, 0, q) {
142         int u = Q[i].id;
143         w = lower_bound(a+1, a+1+m, Q[i].w) - a;
144         vc[u].pb(ques_t(w, i));
145     }
146     
147     dfs(1, 0);
148     
149     rep(i, 0, q) {
150         if (ans[i][0] == -1)
151             puts("0");
152         else
153             printf("%d %d\n", ans[i][1], ans[i][0]);
154     }
155 }
156 
157 int main() {
158     ios::sync_with_stdio(false);
159     #ifndef ONLINE_JUDGE
160         freopen("data.in", "r", stdin);
161         freopen("data.out", "w", stdout);
162     #endif
163     
164     int t;
165     int u, v1, v2;
166     
167     scanf("%d", &t);
168     while (t--) {
169         init();
170         scanf("%d", &n);
171         rep(i, 1, n+1) {
172             scanf("%d", &W[i]);
173             a[m++] = W[i];
174         }
175         int k;
176         scanf("%d", &k);
177         rep(i, 0, k) {
178             scanf("%d %d %d", &u, &v1, &v2);
179             son[u][0] = v1;
180             son[u][1] = v2;
181         }
182         solve();
183     }
184     
185     #ifndef ONLINE_JUDGE
186         printf("time = %d.\n", (int)clock());
187     #endif
188     
189     return 0;
190 }

主席树。

  1 /* 4605 */
  2 #include <iostream>
  3 #include <sstream>
  4 #include <string>
  5 #include <map>
  6 #include <queue>
  7 #include <set>
  8 #include <stack>
  9 #include <vector>
 10 #include <deque>
 11 #include <algorithm>
 12 #include <cstdio>
 13 #include <cmath>
 14 #include <ctime>
 15 #include <cstring>
 16 #include <climits>
 17 #include <cctype>
 18 #include <cassert>
 19 #include <functional>
 20 #include <iterator>
 21 #include <iomanip>
 22 using namespace std;
 23 //#pragma comment(linker,"/STACK:102400000,1024000")
 24 
 25 #define sti                set<int>
 26 #define stpii            set<pair<int, int> >
 27 #define mpii            map<int,int>
 28 #define vi                vector<int>
 29 #define pii                pair<int,int>
 30 #define vpii            vector<pair<int,int> >
 31 #define rep(i, a, n)     for (int i=a;i<n;++i)
 32 #define per(i, a, n)     for (int i=n-1;i>=a;--i)
 33 #define clr                clear
 34 #define pb                 push_back
 35 #define mp                 make_pair
 36 #define fir                first
 37 #define sec                second
 38 #define all(x)             (x).begin(),(x).end()
 39 #define SZ(x)             ((int)(x).size())
 40 // #define lson            l, mid, rt<<1
 41 // #define rson            mid+1, r, rt<<1|1
 42 
 43 const int INF = 1e9+5;
 44 const int maxn = 1e5+5;
 45 const int maxm = maxn * 30;
 46 int a[maxn], W[maxn];
 47 int T[maxn];
 48 int lson[maxm], rson[maxm];
 49 int lc[maxm], rc[maxm];
 50 int n, m, tot;
 51 int son[maxn][2];
 52 
 53 void init() {
 54     tot = m = 0;
 55     memset(son, 0, sizeof(son));
 56 }
 57 
 58 int Build(int l, int r) {
 59     int rt = tot++;
 60     
 61     lc[rt] = rc[rt] = 0;
 62     
 63     if (l == r)
 64         return rt;
 65     
 66     int mid = (l + r) >> 1;
 67     
 68     lson[rt] = Build(l, mid);
 69     rson[rt] = Build(mid+1, r);
 70     
 71     return rt;
 72 }
 73 
 74 int Insert(int rt, int p, int lval, int rval) {
 75     int nrt = tot++, ret = nrt;
 76     int l = 0, r = m - 1, mid;
 77     
 78     lc[nrt] = lc[rt] + lval;
 79     rc[nrt] = rc[rt] + rval;
 80     while (l < r) {
 81         mid = (l + r) >> 1;
 82         if (p <= mid) {
 83             lson[nrt] = tot++;
 84             rson[nrt] = rson[rt];
 85             nrt = lson[nrt];
 86             rt = lson[rt];
 87             r = mid;
 88         } else {
 89             lson[nrt] = lson[rt];
 90             rson[nrt] = tot++;
 91             nrt = rson[nrt];
 92             rt = rson[rt];
 93             l = mid + 1;
 94         }
 95         lc[nrt] = lc[rt] + lval;
 96         rc[nrt] = rc[rt] + rval;
 97     }
 98     
 99     return ret;
100 }
101 
102 pii Query(int rt, int L, int R, int l, int r) {
103     if (L==l && R==r)
104         return mp(lc[rt], rc[rt]);
105     
106     int mid = (l + r) >> 1;
107     
108     if (R <= mid) {
109         return Query(lson[rt], L, R, l, mid);
110     } else if (L > mid) {
111         return Query(rson[rt], L, R, mid+1, r);
112     } else {
113         pii ltmp = Query(lson[rt], L, mid, l, mid);
114         pii rtmp = Query(rson[rt], mid+1, R, mid+1, r);
115         return mp(ltmp.fir+rtmp.fir, ltmp.sec+rtmp.sec);
116     }
117 }
118 
119 void solve() {
120     a[m++] = INF;
121     a[m++] = 0;
122     sort(a, a+m);
123     m = unique(a, a+m) - a;
124     
125     T[1] = Build(0, m-1);
126     queue<int> Q;
127     int u, v;
128     
129     Q.push(1);
130     while (!Q.empty()) {
131         u = Q.front();
132         Q.pop();
133         if (son[u][0]) {
134             int wid = lower_bound(a, a+m, W[u]) - a;
135             T[son[u][0]] = Insert(T[u], wid, 1, 0);
136             T[son[u][1]] = Insert(T[u], wid, 0, 1);
137             Q.push(son[u][0]);
138             Q.push(son[u][1]);
139         }
140     }
141     
142     int q;
143     int w;
144     int n2, n7;
145     
146     scanf("%d", &q);
147     while (q--) {
148         scanf("%d %d", &v, &w);
149         int wid = lower_bound(a, a+m, w) - a;
150         if (a[wid] == w) {
151             // check hit w
152             pii eq = Query(T[v], wid, wid, 0, m-1);
153             if (eq.fir+eq.sec > 0) {
154                 puts("0");
155                 continue;
156             }
157         }
158         
159         pii lt = Query(T[v], 0, wid-1, 0, m - 1);
160         pii gt = Query(T[v], wid, m-1, 0, m - 1);
161         
162         n2 = (gt.fir+gt.sec) + (lt.fir+lt.sec) * 3;
163         n7 = lt.sec;
164         
165         printf("%d %d\n", n7, n2);
166     }
167 }
168 
169 int main() {
170     ios::sync_with_stdio(false);
171     #ifndef ONLINE_JUDGE
172         freopen("data.in", "r", stdin);
173         freopen("data.out", "w", stdout);
174     #endif
175     
176     int t;
177     int u;
178     
179     scanf("%d", &t);
180     rep(tt, 1, t+1) {
181         scanf("%d", &n);
182         init();
183         rep(i, 1, n+1) {
184             scanf("%d", &W[i]);
185             a[m++] = W[i];
186         }
187         int k;
188         scanf("%d", &k);
189         rep(i, 0, k) {
190             scanf("%d", &u);
191             scanf("%d %d", &son[u][0], &son[u][1]);
192         }
193         solve();
194     }
195     
196     #ifndef ONLINE_JUDGE
197         printf("time = %d.\n", (int)clock());
198     #endif
199     
200     return 0;
201 }

数据发生器。

 1 from copy import deepcopy
 2 from random import randint, shuffle
 3 import shutil
 4 import string
 5 
 6 
 7 def GenDataIn():
 8     with open("data.in", "w") as fout:
 9         t = 1
10         bound = 10**5 / 2
11         fout.write("%d\n" % (t))
12         for tt in xrange(t):
13             m = randint(100, 200)
14             n = 2 * m + 1
15             fout.write("%d\n" % (n))
16             L = []
17             for i in xrange(n):
18                 w = randint(1, bound)
19                 L.append(w)
20             fout.write(" ".join(map(str, L)) + "\n")
21             ust = [1]
22             vst = range(2, n+1)
23             fout.write("%d\n" % (m))
24             for i in xrange(m):
25                 idx = randint(0, len(ust)-1)
26                 u = ust[idx]
27                 ust.remove(u)
28                 V = []
29                 for j in xrange(2):
30                     idx = randint(0, len(vst)-1)
31                     v = vst[idx]
32                     V.append(v)
33                     ust.append(v)
34                     vst.remove(v)
35                 fout.write("%d %d %d\n" % (u, V[0], V[1]))
36             q = randint(n/2, n)
37             fout.write("%d\n" % (q))
38             for i in xrange(q):
39                 x = randint(1, n)
40                 w = randint(1, bound)
41                 fout.write("%d %d\n" % (x, w))
42                 
43             
44                 
45 def MovDataIn():
46     desFileName = "F:\eclipse_prj\workspace\hdoj\data.in"
47     shutil.copyfile("data.in", desFileName)
48 
49     
50 if __name__ == "__main__":
51     GenDataIn()
52     MovDataIn()