Codeforces-983E (Round483 Div1) NN country 树上倍增+树状数组

最新推荐文章于 2021-07-16 16:40:33 发布
最新推荐文章于 2021-07-16 16:40:33 发布
阅读量509 收藏 1
点赞数
分类专栏： ACM学习 文章标签： 树状数组 树上倍增
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Desico/article/details/80401772
版权

大意：有一棵树，有m条线路，给出q个询问u->v最少通过几条线路到达。
解法：贪心求u和v到lca最少跳几次，然后判最后一次是否通过同一条线路到达。通过倍增法到达logN级别，最后的特判是用树状数组达到logN级别。

对于每一个节点，其子树代表的区间为dfn[v]到dfn[v]+sz[v]，对于需要特判的一组(u,v)，在dfs到u的时候把起点为u的路的终点+1，相当于有一条路从u戳到某一个区间(即某一个节点的子树中)，最后判断v的子树中是否有从u的子树中戳过来的路，如果有那么就存在u->v的一条线路。对于哪些(u,v)是需要特判的呢？因为程序中求的是u和v还差一次才跳到lca的次数a和b，因此如果u或者v本身就是lca那么就不需要特判。

 


#include<iostream>
#include<string>
#include<cstring>
#include<stdlib.h>
#include<cstdio>
#include<stdio.h>
#include<set>
#include<map>
#include<deque>
#include<stack>
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<list>
#include<bitset>
#include<sstream>
using namespace std;
#define lc n << 1
#define rc n << 1 | 1
typedef long long ll;
typedef unsigned long long ull;
const int maxint = ~0U >> 1;


const int maxn = 2e5 + 7;
const int INF = 0x3f3f3f3f;
const int logN = 20;
struct edge {
int next, to;
}e[maxn];


int n, m, q;
int dfn[maxn], dfed[maxn], anc[maxn][logN], dep[maxn], tot;//lca
int head[maxn], cnt;//road
int top[maxn];//the furthest ancestor
int jump[maxn][logN];
bool bad[maxn];
int ans[maxn],tmpcnt[maxn];
vector<int>wayed[maxn];
vector<pair<int, int> >query[maxn];
inline void add(int x, int y) 
{
e[cnt].to = y;
e[cnt].next = head[x];
head[x] = cnt++;
}
void dfs1(int n, int fa = -1)
{
dfn[n] = ++tot;
dep[n] = (fa == -1 ? 0 : dep[fa] + 1);
anc[n][0] = fa;
for (int i = 1;i < logN;i++)
{
if (anc[n][i - 1] == -1)
anc[n][i] = -1;
else
anc[n][i] = anc[anc[n][i - 1]][i - 1];
}
for (int i = head[n];~i;i = e[i].next)
{
int v = e[i].to;
dfs1(v, n);
}
dfed[n] = tot;
}
bool is_ancestor(int u, int v)//if u is the ancestor of v
{
if (u == -1)return true;
if (v == -1)return false;
return dfn[u] <= dfn[v] && dfed[u] >= dfed[v];
}
int lca(int u, int v)
{
if (is_ancestor(u, v))return u;
if (is_ancestor(v, u))return v;
for (int i = logN - 1;i >= 0;i--)
{
if (!is_ancestor(anc[u][i], v))
u = anc[u][i];
}
return anc[u][0];
}
int calc_top(int n)//求一次最远跳到的点top
{
for (int i = head[n];~i;i = e[i].next)
{
top[n] = min(top[n], calc_top(e[i].to));
}
return top[n];
}
void dfs2(int n)//倍增往上跳
{
jump[n][0] = top[n];
for (int i = 1;i < logN;i++)
jump[n][i] = jump[jump[n][i - 1]][i - 1];
for (int i = head[n];~i;i = e[i].next)
dfs2(e[i].to);
}
bool reachable(int u, int v)
{
int _lca = lca(u, v);
for (int i = logN - 1;i >= 0;i--)
{
if (!is_ancestor(jump[u][i], _lca))
u = jump[u][i];
if (!is_ancestor(jump[v][i], _lca))
v = jump[v][i];
}
u = jump[u][0], v = jump[v][0];
return is_ancestor(u, _lca) && is_ancestor(v, _lca);
}
int calc_cnt(int& u, int v)
{
int cnt = 0;
for (int i = logN - 1;i >= 0;i--)
{
if (!is_ancestor(jump[u][i], v))
{
u = jump[u][i];
cnt += (1 << i);
}
}
return cnt;
}


int a[maxn];
inline int lowbit(int x)
{
return x & -x;
}
void add(int x)
{
while (x <= n)
{
a[x]++;
x += lowbit(x);
}
}
int sum(int x)
{
int ret = 0;
while (x > 0)
{
ret += a[x];
x -= lowbit(x);
}
return ret;
}
void solve(int n)
{
for (auto qry : query[n])
{
int v = qry.first;
tmpcnt[qry.second] += sum(dfed[v]) - sum(dfn[v] - 1);
}
for (auto ed : wayed[n])
add(dfn[ed]);
for (int i = head[n];~i;i = e[i].next)
solve(e[i].to);
for (auto qry : query[n])
{
int v = qry.first, id = qry.second;
tmpcnt[id] -= sum(dfed[v]) - sum(dfn[v] - 1);
if (tmpcnt[id] <= -1)
ans[id]--;
}
}
int main()
{
int u, v, _lca, tmp;
memset(head, -1, sizeof head);
memset(anc, -1, sizeof anc);
memset(jump, -1, sizeof jump);
scanf("%d", &n);
for (int i = 2;i <= n;i++)
{
scanf("%d", &tmp);
add(tmp, i);
}
dfs1(1);//precalc the lca
for (int i = 1;i <= n;i++)
top[i] = i;
scanf("%d", &m);
for (int i = 0;i < m;i++)
{
scanf("%d%d", &u, &v);
wayed[u].push_back(v);
wayed[v].push_back(u);
_lca = lca(u, v);
if (dep[top[v]] > dep[_lca])
top[v] = _lca;
if (dep[top[u]] > dep[_lca])
top[u] = _lca;
}
calc_top(1);
dfs2(1);

#include<iostream>
#include<string>
#include<cstring>
#include<stdlib.h>
#include<cstdio>
#include<stdio.h>
#include<set>
#include<map>
#include<deque>
#include<stack>
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<list>
#include<bitset>
#include<sstream>
using namespace std;
#define lc n << 1
#define rc n << 1 | 1
typedef long long ll;
typedef unsigned long long ull;
const int maxint = ~0U >> 1;


const int maxn = 2e5 + 7;
const int INF = 0x3f3f3f3f;
const int logN = 20;
struct edge {
int next, to;
}e[maxn];


int n, m, q;
int dfn[maxn], dfed[maxn], anc[maxn][logN], dep[maxn], tot;//lca
int head[maxn], cnt;//road
int top[maxn];//the furthest ancestor
int jump[maxn][logN];
bool bad[maxn];
int ans[maxn],tmpcnt[maxn];
vector<int>wayed[maxn];
vector<pair<int, int> >query[maxn];
inline void add(int x, int y) 
{
    e[cnt].to = y;
    e[cnt].next = head[x];
    head[x] = cnt++;
}
void dfs1(int n, int fa = -1)
{
    dfn[n] = ++tot;
    dep[n] = (fa == -1 ? 0 : dep[fa] + 1);
    anc[n][0] = fa;
    for (int i = 1;i < logN;i++)
    {
        if (anc[n][i - 1] == -1)
            anc[n][i] = -1;
        else
            anc[n][i] = anc[anc[n][i - 1]][i - 1];
    }
    for (int i = head[n];~i;i = e[i].next)
    {
        int v = e[i].to;
        dfs1(v, n);
       }
    dfed[n] = tot;
}
bool is_ancestor(int u, int v)//if u is the ancestor of v
{
    if (u == -1)return true;
    if (v == -1)return false;
    return dfn[u] <= dfn[v] && dfed[u] >= dfed[v];
}
int lca(int u, int v)
{
    if (is_ancestor(u, v))return u;
    if (is_ancestor(v, u))return v;
    for (int i = logN - 1;i >= 0;i--)
    {
        if (!is_ancestor(anc[u][i], v))
        u = anc[u][i];
    }
    return anc[u][0];
}
int calc_top(int n)//求一次最远跳到的点top
{
    for (int i = head[n];~i;i = e[i].next)
    {
        top[n] = min(top[n], calc_top(e[i].to));
    }
    return top[n];
}
void dfs2(int n)//倍增往上跳
{
    jump[n][0] = top[n];
    for (int i = 1;i < logN;i++)
        jump[n][i] = jump[jump[n][i - 1]][i - 1];
    for (int i = head[n];~i;i = e[i].next)
        dfs2(e[i].to);
}
bool reachable(int u, int v)
{
    int _lca = lca(u, v);
    for (int i = logN - 1;i >= 0;i--)
    {
        if (!is_ancestor(jump[u][i], _lca))
            u = jump[u][i];
        if (!is_ancestor(jump[v][i], _lca))
            v = jump[v][i];
    }
    u = jump[u][0], v = jump[v][0];
    return is_ancestor(u, _lca) && is_ancestor(v, _lca);
}
int calc_cnt(int& u, int v)
{
    int cnt = 0;
    for (int i = logN - 1;i >= 0;i--)
    {
        if (!is_ancestor(jump[u][i], v))
        {
            u = jump[u][i];
            cnt += (1 << i);
        }
    }
    return cnt;
}


int a[maxn];
inline int lowbit(int x){return x & -x;}
void add(int x)
{
    while (x <= n)
    {
        a[x]++;
        x += lowbit(x);
    }
}
int sum(int x)
{
    int ret = 0;
    while (x > 0)
    {
        ret += a[x];
        x -= lowbit(x);
    }
    return ret;
}
void solve(int n)
{
    for (auto qry : query[n])
    {
        int v = qry.first;
        tmpcnt[qry.second] += sum(dfed[v]) - sum(dfn[v] - 1);
    }
    for (auto ed : wayed[n])
    add(dfn[ed]);
    for (int i = head[n];~i;i = e[i].next)
        solve(e[i].to);
    for (auto qry : query[n])
    {
        int v = qry.first, id = qry.second;
        tmpcnt[id] -= sum(dfed[v]) - sum(dfn[v] - 1);
        if (tmpcnt[id] <= -1)
            ans[id]--;
    }
   }
int main()
{
    int u, v, _lca, tmp;
    memset(head, -1, sizeof head);
    memset(anc, -1, sizeof anc);
    memset(jump, -1, sizeof jump);
    scanf("%d", &n);
    for (int i = 2;i <= n;i++)
    {
        scanf("%d", &tmp);
        add(tmp, i);
    }
    dfs1(1);//precalc the lca
    for (int i = 1;i <= n;i++)
        top[i] = i;
    scanf("%d", &m);
    for (int i = 0;i < m;i++)
    {
        scanf("%d%d", &u, &v);
        wayed[u].push_back(v);
        wayed[v].push_back(u);
        _lca = lca(u, v);
        if (dep[top[v]] > dep[_lca])
            top[v] = _lca;
        if (dep[top[u]] > dep[_lca])
            top[u] = _lca;
    }
    calc_top(1);
    dfs2(1);
    scanf("%d", &q);
    for(int i = 0;i < q;i++)
    {
        scanf("%d%d", &u, &v);
        if (!reachable(u, v))bad[i] = true;
        _lca = lca(u, v);
        ans[i] = calc_cnt(u, _lca) + calc_cnt(v, _lca);
        if (u != _lca)ans[i]++;
        if (v != _lca)ans[i]++;
        if (u != _lca && v != _lca)query[u].push_back({ v,i });
    }
    solve(1);
    for (int i = 0;i < q;i++)
    {
        printf("%d\n", bad[i] ? -1 : ans[i]);
    }
    return 0;
}


scanf("%d", &q);
for(int i = 0;i < q;i++)
{
scanf("%d%d", &u, &v);
if (!reachable(u, v))bad[i] = true;
_lca = lca(u, v);
ans[i] = calc_cnt(u, _lca) + calc_cnt(v, _lca);
if (u != _lca)ans[i]++;
if (v != _lca)ans[i]++;
if (u != _lca && v != _lca)query[u].push_back({ v,i });
}
solve(1);
for (int i = 0;i < q;i++)
{
printf("%d\n", bad[i] ? -1 : ans[i]);
}
return 0;
}

 

Desico
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
在树状数组上倍增（广东省赛2023）
C20181503csy的博客
06-30 178
假设从x出发，以找区间的左端点l为例。那么我们可以提前计算出在前x个位置颜色集合中颜色出现的总次数（如果为x则l肯定为1），然后直接减去color数组中nxt位置的贡献（因为nxt中未计算的部分一定为2的幂次处的贡献），就可以计算出区间[nxt+1,x]中出现的颜色次数。根据上面的描述，最后的结果l’应该是最靠近x的使得[l’+1,x]中出现的颜色次数比区间长度小的点，所以真正的l应该是l’+2。对于询问，本质上就是要找到一个尽量长的包含x的区间，使得区间中出现的目标颜色总次数与区间长度相等。
codeforces 932D. Tree(树上倍增)
the best 的博客
07-28 135
题意： 给出一个初始点只有1的树，权值为0，有两种操作： 1.1.1. 往树中加入一个节点 2.查询从某个节点起，往祖先方向能找到的最长非降子序列。要求：权值和不能超过xxx ,并且后一个点是前一个点的祖先，且必须是最近的，权值≥\geq≥ 前一个点的祖先。 操作强制在线。 题解： 像这种往祖先方向查询的操作，可以考虑倍增法。 设dp[i][j]dp[i][j]dp[i][j]​ 表示节点iii 向上找长度为2j2^j2j 的非降子序列后所处的位置 设sum[i][j]sum[i][j]sum[i][j]
[CF983E]NN country
weixin_34233421的博客
06-04 131
题意：给一棵树，有许多条巴士线路$(a_i,b_i)$（巴士在路径上每个点都会停车），多次询问从一点到另一点最少要坐多少次巴士 首先dfs一遍预处理出一个点向上坐$2^k$次巴士能到的最浅点，于是我们能很快地查询一个点往上走到另一个点最少要坐多少次巴士 对于询问$(u,v)$，我们肯定是贪心地坐车，每次坐尽可能远，路径是$u\rightarrow lca_{u,v}\rightarrow v$...
【Codeforces 983E】NN country 题解
蒟
05-26 445
题面：传送门 &amp;nbsp;&amp;nbsp;&amp;nbsp;&amp;nbsp;一道神奇的题目，很容易想到树上倍增，如果xxx是yyy的祖先或yyy是xxx的祖先，直接倍增跑一遍即可，否则，我们会发现x⇒yx⇒yx\Rightarrow y的答案其实可以分为(x⇒lcax,y)+(lcax,y⇒y)(x⇒lcax,y)+(lcax,y⇒y)(x\Rightarrow lca_{x,y})+(lca_{x,y}...
Codeforces983E. NN country
a2850363633的博客
05-17 202
新鲜出炉！ $n \leq 200000$的树，给$m \leq 200000$条链，$q \leq 200000$个询问，每次问一条询问链最少用m条中的几条给定链覆盖其所有边，可能无解。 首先确定一种贪心策略：$low_x$表示$x$能通过一条链跳到的最高的点（这里吐槽一下，题解描述时的树是反过来的，树根在地上！反人类啊，然后他后面所有的“高”“矮”描述全是反过来的！！！无良出题人...
Codeforces-Solution
05-03
"Codeforces-Solution"很可能是一个包含针对Codeforces上问题的解题代码库，主要使用C++语言，部分可能涉及cpp14标准。 在这个压缩包中，"Codeforces-Solution-master"可能是这个代码库的主分支或者最新版本，这...
a2oj-codeforces-div2c-ladder-solutions
05-09
a2oj-codeforces-div2c-ladder-solutions 我的A2online梯子裁判解决方案div2c codeforces
PDF-CodeForces-Problems:PDF文件中的CodeForces问题
05-12
PDF-CodeForces问题 PDF文件中的CodeForces问题 利用CodeForces提取在这个库中的文件 ，文件夹名代表来自CodeForces网址contests`的ID，每个文件夹包含比赛的问题。 有关CodeForces竞赛的疯狂事实 CodeForces上有937...
Educational Codeforces Round 83 (Rated for Div. 2) D
01-20
C(n-2,i-1)*C(j-1,n-2)*(i-1) __ j: n-1 -> m 我们发现内层循环，每次只是j加一，我们就可以只用一次组合数剩下的用差量表示 对于外层循环同理 只有(i-1) * C(n-2,i-1) i会每次加一。我们也只算一次剩下的用差量...
leetcode卡-codeforces-stats:codeforces-stats
07-06
leetcode卡Codeforces 统计 一个工具可以帮助您显示 CF 统计信息。 灵感来自 . 范围 范围 默认 必需的 处理 空值 真的 贡献 错误的 错误的 朋友们 错误的 错误的 主题 默认 错误的 主题选择：默认、黑暗 例子 默认 ...
Codeforces 983E NN country 倍增+树状数组
beginend
05-24 599
题意 给一棵n个节点的树和m条链，有q次询问，每次询问给出两个点x和y，问最少选出多少条链使得x到y路径上的每条边都至少被覆盖一次。 n,m,q≤2∗105n,m,q≤2∗105n,m,q\le2*10^5 分析 考虑先预处理每个点往上跳一步最多可以跳到哪里，然后先贪心从两个端点往lca跳，求出跳到某个离lca最近且不为lca的点，设为a和b。 接下来分两种情况，若a可以一步到达b，...
POJ2182 Lost Cows（树状数组、二分、倍增）
m0_51864047的博客
07-16 262
Lost Cows 题目链接 Description N (2 <= N <= 8,000) cows have unique brands in the range 1…N. In a spectacular display of poor judgment, they visited the neighborhood ‘watering hole’ and drank a few too many beers before dinner. When it was time to line u
Codeforces483Div1 983E NN country
氧化钠的博客
05-26 460
题意： 给出一颗N个点的树，树上有M条链，需要满足Q次询问： 每次询问给出两个点(u,v)(u,v)(u,v)，求从u出发，只通过链来行走（即当前位于某条链的某一个点上，就可以移动到这条链的任意一个点上，这定义为一步），求到达v点的最小步数。链可能会重复覆盖某些点，也可能不连通，若无法到达输出-1 N,Q,M≤105N,Q,M≤105N,Q,M\leq 10^5 分析： 首先，有...
codeforces 587C(树上的倍增算法)
playwfun的专栏
11-09 1289
分析： 同LCA相似的思路，维护st[ i ][ j ] 代表从节点i往上（不包含i）走2^j个节点前十个最小值集合，这样一直倍增着往上走秩序logn次集合合并即可。 #include #define fst first #define snd second #define ALL(a) a.begin(), a.end() #define clr(a, x) memset(a, x, s
【学习】彻底理解树状数组
yhf_2015
12-24 2847
前言： 可能是因为学习了很多高级数据结构的缘故，突然感觉好像明白了树状数组，重新总结一下。 本文通过从根源深处挖掘树状数组所解决的问题，深刻的理解树状数组的操作本质，若要系统的研究树状数组，建议学习一下“二进制分解”“倍增”的概念。 考虑到初学者，文章写的比较长，废话比较多，还望耐心的看下去，相信你也能有新的收获。 温馨提示：文章中的代码仅供参考思路，不保证100%正确，使用时请根据原题情况
CF983E NN country
某蒟蒻的博客
05-19 3408
NN country 题目描述 传送门：http://codeforces.com/contest/983/problem/E 题解 首先有一个很显然的贪心策略，我们对于每个节点预处理出从它出发向上乘一次车最远能到哪。 对于一次询问，两个点x,y，我们先让这两个点贪心地往lca方向跑。 这样x点跑了a次到了lx点，y点跑了b次到了rx点。 如果lx到rx可以乘一次车到达，那么答案是a...
CF983E NN country（倍增,差分）
Aaronliu17008的博客
07-30 206
题意 给定一棵树和若干条路线，每条路线相当于树上 x,y 之间的路径，途径路径上的每个点 给出若干个询问，每次询问从 u 到 v 至少需要利用几条路线 N,M,Q≤200000 题解 构建倍增数组g[i][j]表示从i点向上经过j条线路能到达的深度最小的点。 所以对于每一对询问的x,y,我们贪心地把它们提到深度大于等于lca地最大的点。记为x‘，y'然后判断是否有路径经过x’...
Codeforces 983E NN country 倍增 树状数组
u011535421的博客
02-14 295
http://codeforces.com/contest/983/problem/E   E. NN country time limit per test 3 seconds memory limit per test 256 megabytes input standard input output standard output In the NN country, ...
Codeforces 827D (Round #423 Div. 1) D. Best Edge Weight 树上倍增
J. Cole
08-18 632
D. Best Edge Weight time limit per test 2 seconds memory limit per test 256 megabytes input standard input output standard output You are given a connected wei
CodeForces - 982C 树形DP
最新发布
07-25
引用\[1\]中提到了一种树形动态规划的方法来解决CodeForces - 982C问题。在这个问题中，subtree指的是子连通块，而不是子树。为了使cnt_white - cnt_black尽可能大，可以使用两次树形动态规划来求解。第一次是自底向上的过程，维护一个dp数组，表示以每个节点为根的子树中的最大连通块。第二次是自顶向下的过程，处理自底向上过程中无法包含的树链所代表的子树。在第二次遍历中，需要维护一个sum变量，用于存储树链所代表的子树的贡献。根据ans\[u\]的正负，决定是否能对相邻的子节点做出贡献。如果ans\[u\]为正，则减去dp\[v\]就是树链所代表的子树的权值。最终，ans\[u\]代表包含节点u在内的子连通块的最大权值。\[1\] 问题: CodeForces - 982C 树形DP是什么问题？如何解决？ 回答: CodeForces - 982C是一个树形动态规划问题。在这个问题中，需要求解子连通块的最大权值和，使得cnt_white - cnt_black尽可能大。解决这个问题的方法是使用两次树形动态规划。第一次是自底向上的过程，维护一个dp数组，表示以每个节点为根的子树中的最大连通块。第二次是自顶向下的过程，处理自底向上过程中无法包含的树链所代表的子树。在第二次遍历中，需要维护一个sum变量，用于存储树链所代表的子树的贡献。根据ans\[u\]的正负，决定是否能对相邻的子节点做出贡献。最终，ans\[u\]代表包含节点u在内的子连通块的最大权值。\[1\] #### 引用[.reference_title] - *1* *2* [CodeForces - 1324F Maximum White Subtree(树形dp)](https://blog.csdn.net/qq_45458915/article/details/104831678)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值