https://codeforc.es/gym/102059/problem/A
题意 给出一棵树。
询问u, c, m
: 将结点u到根节点路径上的边都染色成c,询问染色边数为m的颜色有多少种。
题解 每次都是从一个结点到根,所以可以对这颗树进行树链剖分(heavy-light decompose)
维护
has[col]
代表颜色col有多少条边。
cnt[m]
边数为m的颜色有多少个。
dfn[v]
:结点v的dfs编号。
对于每条链的头维护一个dfn的单调递增栈。
栈中的内容为(dfn,c)
,从dfn
处用颜色c
染色到链头。
对于一个染色操作,模拟树链跑到根结点的过程。
如果当前的结点dfn
大于栈头的dfn
,那就要将栈头的这个颜色的影响去掉,并弹出。
如果当前的结点的dfn
小于栈头的dfn
,那就将当前dfn
到链头的之前颜色的影响去掉。
将dfn到链头染色为c,然后加入单调栈中。
代码 【树链剖分|【Gym 102059A】Coloring Roads(树链剖分+单调栈)】官方std
#include
using namespace std;
using lint = long long;
using pi = pair;
const int MAXN = 200005;
int n, c, q, dfn[MAXN], chn[MAXN], sz[MAXN], par[MAXN], piv;
vector gph[MAXN];
void dfs0(int x){
sz[x] = 1;
for(auto &i : gph[x]){
gph[i].erase(find(gph[i].begin(), gph[i].end(), x));
par[i] = x;
dfs0(i);
sz[x] += sz[i];
}
}void dfs1(int x){
dfn[x] = ++piv;
if(gph[x].empty()) return;
chn[gph[x][0]] = chn[x];
dfs1(gph[x][0]);
for(int i=1;
i sz[b];
});
}
dfs1(1);
while(q--){
int v, c;
scanf("%d %d",&v,&c);
while(v){
int ch = chn[v];
int prv = dfn[ch] - 1;
while(!stk[ch].empty() && stk[ch].back().first <= dfn[v]){
auto val = stk[ch].back();
stk[ch].pop_back();
change_color(val.second, -(val.first - prv));
prv = val.first;
}
if(!stk[ch].empty()){
auto val = stk[ch].back();
change_color(val.second, -(dfn[v] - prv));
}
change_color(c, dfn[v] - dfn[ch] + 1);
stk[ch].emplace_back(dfn[v], c);
v = par[ch];
}
int m;
scanf("%d",&m);
printf("%d\n", cnt[m] - (has[c] == m) + (has[c] == m + 1));
}
}
推荐阅读
- 树链剖分|codeforces 208 E. Blood Cousins (dsu on the tree)
- 树链剖分|牛客练习赛51 F-ABCBA(树链剖分,线段树,状态转移)
- 单调栈|2020杭电多校第一场(解题报告)