UOJ #58 糖果公园

Candyland 有一座糖果公园，公园里不仅有美丽的风景、好玩的游乐项目，还有许多免费糖果的发放点，这引来了许多贪吃的小朋友来糖果公园玩。

糖果公园的结构十分奇特，它由 $n$

糖果公园所发放的糖果种类非常丰富，总共 $m$

来到公园里游玩的游客都不喜欢走回头路，他们总是从某个特定的游览点出发前往另一个特定的游览点，并游览途中的景点，这条路线一定是唯一的。他们经过每个游览点，都可以品尝到一颗对应种类的糖果。

大家对不同类型的糖果的喜爱程度都不尽相同。根据游客们的反馈打分，我们得到了糖果的美味指数，第 $i$

当然，公园中每个糖果发放点所发放的糖果种类不一定是一成不变的。有时，一些糖果点所发放的糖果种类可能会更改（也只会是 $m$

糖果公园的工作人员小 A 接到了一个任务，那就是根据公园最近的数据统计出每位游客游玩公园的愉悦指数。但数学不好的小 A 一看到密密麻麻的数字就觉得头晕，作为小 A 最好的朋友，你决定帮他一把。

输入格式

第一行包含三个正整数 $n, m, q$

第二行包含 $m$

第三行包含 $n$

第四行到第 $n + 2$

第 $n + 3$

接下来 $q$

若 $t$

输出格式

按照输入的先后顺序，对于每个 $t$

样例一

input

output

限制与约定

对于所有的数据， $1 \leq v_{i}, w_{i} \leq 10^{6}$

测试点编号	$n$	$m$	$q$	其它限制
1	$\leq 20$	$\leq 20$	$\leq 20$	无
2	$\leq 2000$	$\leq 2000$	$\leq 2000$
3	$\leq 10000$	$\leq 10000$	$\leq 10000$
4	$\leq 80000$	$\leq 100$	$\leq 80000$	没有修改操作；给出的图构成一条链
5	$\leq 90000$	$\leq 100$	$\leq 90000$	没有修改操作；给出的图构成一条链
6	$\leq 80000$	$\leq 80000$	$\leq 80000$	没有修改操作
7	$\leq 90000$	$\leq 90000$	$\leq 90000$	没有修改操作
8	$\leq 80000$	$\leq 80000$	$\leq 80000$	给出的图构成一条链
9	$\leq 90000$	$\leq 90000$	$\leq 90000$	给出的图构成一条链
10	$\leq 100000$	$\leq 100000$	$\leq 100000$	无

祝大家一遍 AC，求不虐萌萌哒测评机！

时间限制： $8 s$

空间限制： $512 MB$

真是卡OJ神题，我这道题加起来大概卡了UOJ有10min

树上带修莫队，按照DFS序分块或者王室联邦都行，好像王室联邦会更快

移动端点就是修改两条链，只要不断的往LCA爬，将路径上的点存在性取反就行了，不用讨论，但要记得LCA只翻转一次

记得排序的时候按照所属块排序，左右端点都是，不要写出右端点直接比较这种东西来了

  1 #include <iostream>
  2 #include <cstdlib>
  3 #include <cstdio>
  4 #include <algorithm>
  5 #include <string>
  6 #include <cstring>
  7 #include <cmath>
  8 #include <map>
  9 #include <stack>
 10 #include <set>
 11 #include <vector>
 12 #include <queue>
 13 #include <time.h>
 14 #define eps 10e-7
 15 #define INF 0x3f3f3f3f
 16 #define MOD 1000000007
 17 #define rep0(j,n) for(int j=0;j<n;++j)
 18 #define rep1(j,n) for(int j=1;j<=n;++j)
 19 #define pb push_back
 20 #define set0(n) memset(n,0,sizeof(n))
 21 #define ll long long
 22 #define iter(i,v) for(edge *i = head[v];i;i = i->nxt)
 23 #define max(a,b) (a>b?a:b)
 24 #define min(a,b) (a<b?a:b)
 25 //#define OJ
 26 using namespace std;
 27 char BUF[10000000], *buf;
 28 int read() {
 29     int x = 0;
 30     while (!isdigit(*buf)) { buf++; }
 31     while (isdigit(*buf)) { x = x * 10 + *buf++ - '0'; }
 32     return x;
 33 }
 34 char get_ch() {
 35     char c = getchar();
 36     while (!isalpha(c)) c = getchar();
 37     return c;
 38 }
 39 const int MAXINT = 100010;
 40 const int MAXNODE = 100010;
 41 ll ans, w[MAXNODE], ps_w[MAXINT], qa[MAXINT];
 42 int st[18][MAXNODE*4]={},seq[MAXNODE*4]={},cnt_seq=0,fst[MAXNODE]={},max_p2[262145]={},no[MAXNODE],stk[MAXNODE],top=0,cnt_bl=1;
 43 int dfn[MAXNODE]={}, cnt_dfn=0, bl[MAXNODE]={}, n=0, m=0, v[MAXINT]={}, c[MAXNODE]={}, dep[MAXNODE]={}, in_ans[MAXNODE]={}, fa[MAXNODE]={}, cnt_eg, sz_b, q, cnt_cd[MAXINT]={}, cnt_mdf = -1, cnt_qr;
 44 struct edge {
 45     int u, v;
 46     edge *nxt;
 47     edge(int _u, int _v, edge *_nxt) {
 48         u = _u; v = _v; nxt = _nxt;
 49     }
 50     edge() {}
 51 }*head[MAXNODE], mp[MAXNODE * 2];
 52 struct op {
 53     int tp, x, y, lstc, no;
 54     op(int _x, int _y) {
 55         tp = 1;
 56         x = _x;
 57         y = _y;
 58     }
 59     op(int lst, int _x, int _y, int _no) {
 60         lstc = lst;
 61         tp = 0;
 62         //if (dfn[_x] > dfn[_y]) swap(_x, _y);
 63         x = _x;
 64         y = _y;
 65         no = _no;
 66     }
 67     op() {}
 68 }mdf[MAXINT], qr[MAXINT];
 69 int cmp(const op &a, const op &b) {
 70     if(bl[a.x]==bl[b.x]&&bl[a.y]==bl[b.y]) {if(bl[a.x]&1)return a.lstc<b.lstc; else return a.lstc>b.lstc;}
 71     if(bl[a.x]==bl[b.x])  {if(bl[a.x]&1) return bl[a.y]<bl[b.y]; else return bl[a.y]>bl[b.y];}
 72     return bl[a.x] < bl[b.x];
 73 }
 74 inline void rev(int);
 75 inline void add(int);
 76 inline void remove(int);
 77 inline void add(op&);
 78 inline void remove(op&);
 79 inline void add_edge(int, int);
 80 inline void get_st();
 81 inline void get_input();
 82 inline void work();
 83 inline void solve(int,int);
 84 inline int get_lca(int, int);
 85 //inline void dfs(int, int);
 86 inline int dfs(int, int);
 87 int main() {
 88     //freopen("in.txt", "r", stdin);
 89     //freopen("out.tle", "w", stdout);
 90     for(int i=0;i<=17;i++){
 91         for (int j=(1<<i);j<(1<<(i+1));j++){
 92             max_p2[j]=i;
 93         }
 94     }
 95     get_input();
 96     get_st();
 97     work();
 98     //printf("%d\n",clock());
 99     return 0;
100 }
101 inline void rev(int p) {
102     if (in_ans[p]) {
103         in_ans[p] = 0;
104         ans -= w[cnt_cd[c[p]]] * v[c[p]];
105         cnt_cd[c[p]]--;
106     }
107     else {
108         in_ans[p] = 1;
109         cnt_cd[c[p]]++;
110         ans += w[cnt_cd[c[p]]] * v[c[p]];
111     }
112 }
113 inline void solve(int p1,int p2){
114     while(p1!=p2){
115         dep[p1]>dep[p2]?(rev(p1),p1=fa[p1]):(rev(p2),p2=fa[p2]);
116     }
117 }
118 inline void work() {
119     rep0(i,cnt_qr) if(bl[qr[i].x]<bl[qr[i].y]) swap(qr[i].x,qr[i].y);
120     sort(qr, qr + cnt_qr, cmp);
121     int l, r, pl, pr, lca, lca2, pq = -1;
122     for (int i = pq; i > qr[0].lstc; i--) {
123         remove(mdf[i]);
124     }
125     for (int i = pq + 1; i <= qr[0].lstc; i++) {
126         add(mdf[i]);
127     }
128     pl = l = qr[0].x;
129     pr = r = qr[0].y;
130     lca = get_lca(l, r);
131     solve(pl,pr);
132     rev(lca);
133     qa[qr[0].no] = ans;
134     pq = qr[0].lstc;
135     rep1(i, cnt_qr - 1) {
136         for (int j = pq; j > qr[i].lstc; j--) {
137             remove(mdf[j]);
138         }
139         for (int j = pq + 1; j <= qr[i].lstc; j++) {
140             add(mdf[j]);
141         }
142         lca = get_lca(l, r);
143         lca2 = get_lca(qr[i].x, qr[i].y);
144         solve(l,qr[i].x);
145         solve(r,qr[i].y);
146         rev(lca);
147         rev(lca2);
148         l = qr[i].x; r = qr[i].y;
149         pq = qr[i].lstc;
150         qa[qr[i].no] = ans;
151     }
152     rep0(i, cnt_qr) {
153         printf("%lld\n", qa[i]);
154     }
155 }
156 inline void add_edge(int u, int v) {
157     mp[cnt_eg] = edge(u, v, head[u]);
158     head[u] = &mp[cnt_eg++];
159     mp[cnt_eg] = edge(v, u, head[v]);
160     head[v] = &mp[cnt_eg++];
161 }
162 inline void get_input() {
163     int tp, _u, _v;
164     BUF[fread(BUF, 1, 10000000, stdin)]=0;
165     buf = BUF;
166     n = read(); m = read(); q = read();
167     sz_b = ceil(pow(n, 2.0 / 3.0))/2+1;
168     rep1(i, m) v[i] = read();
169     rep1(i, n) w[i] = read();
170     rep1(i, n) ps_w[i] += ps_w[i - 1] + w[i];
171     rep1(i, n - 1) {
172         _u = read(); _v = read();
173         add_edge(_u, _v);
174     }
175     dfs(1, 0);
176     while(top) bl[stk[--top]]=cnt_bl-1;
177     rep1(i, n) c[i] = read();
178     rep0(i, q) {
179         tp = read(); _u = read(); _v = read();
180         if (tp == 1) qr[cnt_qr] = op(cnt_mdf, _u, _v, cnt_qr), cnt_qr++;
181         else mdf[++cnt_mdf] = op(_u, _v);
182     }
183 }
184 /*inline void dfs(int p, int f) {
185     fa[p] = f;
186     dep[p] = dep[f] + 1;
187     no[cnt_dfn]=p;
188     dfn[p] = ++cnt_dfn;
189     bl[p] = dfn[p] / sz_b;
190     fst[p] = cnt_seq;
191     seq[cnt_seq++]=dfn[p];
192     iter(i, p) {
193         if (i->v == f) continue;
194         dfs(i->v, p);
195         seq[cnt_seq++]=dfn[p];
196     }
197 }*/
198 inline int dfs(int p, int f) {
199     int sz=0,bt=top;
200     fa[p] = f;
201     dep[p] = dep[f] + 1;
202     no[cnt_dfn]=p;
203     dfn[p] = ++cnt_dfn;
204     //bl[p] = dfn[p] / sz_b;
205     fst[p] = cnt_seq;
206     seq[cnt_seq++]=dfn[p];
207     iter(i, p) {
208         if (i->v == f) continue;
209         sz+=dfs(i->v, p);
210         seq[cnt_seq++]=dfn[p];
211         if (top-bt>=sz_b){
212             while(top!=bt) bl[stk[--top]]=cnt_bl;
213             cnt_bl++,sz=0;
214         }
215     }
216     stk[top++]=p;
217     return sz+1;
218 }
219 inline void get_st() {
220     rep0(i,cnt_seq) st[0][i]=seq[i];
221     for (int i = 1; i <= 17; i++)
222         for (int j = 0; j < cnt_seq-(1<<i)+1; j++)
223             st[i][j] = min(st[i-1][j],st[i-1][j+(1<<(i-1))]);
224 }
225 inline int get_lca(int x, int y) {
226     x=fst[x];y=fst[y];
227     if(x>y) swap(x,y);
228     int l = y-x+1;
229     return no[min(st[max_p2[l]][x],st[max_p2[l]][y-(1<<max_p2[l])+1])-1];
230 }
231 inline void remove(int p) {
232     ans -= w[cnt_cd[p]] * v[p];
233     cnt_cd[p]--;
234 }
235 inline void add(int p) {
236     cnt_cd[p]++;
237     ans += w[cnt_cd[p]] * v[p];
238 }
239 inline void add(op& cg) {
240     //int f = get_lca(l, r), f1 = get_lca(l, cg.x), f2 = get_lca(r, cg.x);
241     cg.lstc = c[cg.x];
242     c[cg.x] = cg.y;
243     if(in_ans[cg.x]){
244     //if ((f == f1&&f2 == cg.x) || (f == f2&&f1 == cg.x)) {
245         ans -= w[cnt_cd[cg.lstc]] * v[cg.lstc];
246         cnt_cd[cg.lstc]--;
247         cnt_cd[cg.y]++;
248         ans += w[cnt_cd[cg.y]] * v[cg.y];
249     }
250 }
251 inline void remove(op& cg) {
252     //int f = get_lca(l, r), f1 = get_lca(l, cg.x), f2 = get_lca(r, cg.x);
253     c[cg.x] = cg.lstc;
254     //if ((f == f1&&f2 == cg.x) || (f == f2&&f1 == cg.x)) {
255     if(in_ans[cg.x]){
256         ans -= w[cnt_cd[cg.y]] * v[cg.y];
257         cnt_cd[cg.y]--;
258         cnt_cd[cg.lstc]++;
259         ans += w[cnt_cd[cg.lstc]] * v[cg.lstc];
260     }
261 }

posted @ 2017-04-21 20:45 LoveYayoi 阅读(277) 评论(0) 编辑收藏举报

刷新页面返回顶部

弥生三月

ん、大丈夫…。焦らずゆっくりで。弥生はここにいるよ…うん、いる。