Codeforces 165D - Beard Graph (树链剖分 + 树状数组维护)

题意

给一颗 n n n个节点的树,初始时每条边的颜色都是黑色,现在有三种操作:

  1. 将第 i i i条边染黑,保证染色之前这条边为白色;
  2. 将第 i i i条边染白,保证染色之前这条边为黑色;
  3. 查询 x , y x, y x,y之间的最短路径,若 x , y x, y x,y之间的最短路径中有白色边则输出"-1", 否则输出 x , y x,y x,y距离.

做法

题目需要维护树上任意两点之间的边,很自然想到树链剖分,
树链剖分后的序列需要单点修改区间查询,因此可以用树状数组维护.

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 7;
typedef pair<int, int> pii;
#define fi first
#define se second
#define empb emplace_back

int T[N], n, q;
void add(int x, int p) {
    for(; x<=n; x+=x&-x)
        T[x] += p;
}
int sum(int x) {
    int ret = 0;
    for(; x; x-=x&-x)
        ret+=T[x];
    return ret;
}

vector<int> E[N];
int dep[N], sz[N], son[N], fa[N], id[N], top[N];
int idx;
pair<int, int> P[N];
void dfs1(int u, int d) {
    dep[u] = d; sz[u] = 1;
    int &maxs = son[u];
    for(int &v : E[u]) if(v != fa[u]) {
        fa[v] = u; dfs1(v, d + 1);
        sz[u] += sz[v];
        if(sz[v] > sz[maxs])
            maxs = v;
    }
}
void dfs2(int u, int tp) {
    top[u] = tp; id[u] = ++idx;
    if(son[u]) dfs2(son[u], tp);
    for(int &v : E[u])
        if(v != son[u] && v != fa[u])
            dfs2(v, v);
}
int solve(int u, int v) {
    int uu = top[u], vv = top[v], ret = 0;
    while(uu != vv) {
        if(dep[uu] < dep[vv]) swap(u, v), swap(uu, vv);
        if(sum(id[u]) - sum(id[uu] - 1)) return -1;
        ret += dep[u] - dep[uu] + (uu != fa[uu]);
        u = fa[uu], uu = top[u];
    }
    if(dep[u] < dep[v]) swap(u, v);
    if(sum(id[u]) - sum(id[v])) return -1;
    return ret + dep[u] - dep[v];
}

int main() {
#ifdef local
    freopen("in.txt", "r", stdin);
#endif
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);

    cin >> n;
    for(int i = 1, u, v; i < n; i++) {
        cin >> u >> v; P[i] = pii(u, v);
        E[u].empb(v), E[v].empb(u);
    }
    dfs1(1, 1), dfs2(1, 1);
    cin >> q;
    for(int i = 0, op, x, y; i < q; i++) {
        cin >> op;
        if(op == 1) {
            cin >> x; y = P[x].se, x = P[x].fi;
            if(dep[x] < dep[y]) swap(x, y);
            add(id[x], -1);
        } else if(op == 2) {
            cin >> x; y = P[x].se, x = P[x].fi;
            if(dep[x] < dep[y]) swap(x, y);
            add(id[x], 1);
        } else {
            cin >> x >> y;
            cout << solve(x, y) << endl;
        }
    }

    return 0;
}
全部评论

相关推荐

04-12 21:52
南开大学 Java
鼠鼠有点摆,去年边学着没敢投简历,没实习。从1月到现在总共面了五次,四次字节的日常(HR打电话约面试才敢去的),然后一次腾讯的暑期,都是一面挂,其他则是没给面。暑期的岗,4.2才开始海投,前面想着等字节第四次一面后再投,结果挂,而且感觉投晚了。字节投了11个,9个简历挂,剩下2个没动静。阿里全都简历挂,剩下的在&quot;投递简历&quot;。腾讯给了一次面。然后其他大中厂、手机厂什么的都是做完测评or笔试就没下文,打开几个看也是终止流程,感觉剩下的也应该是简历挂了。感觉是简历的原因?项目部分,几次面试,感觉面试官主要就拷问过秒杀这一个点。自己说的时候会尝试把sse那条说成亮点,但除了腾讯面试官问过一下这整个点在业务方面对用户有什么用之类的问题外,其他最多只是问一下sse八股...感觉也许不是很让面试官感兴趣。这个短链接也是无人问津,就被问过一回雪花算法的设计。也许我该拿点评改改,然后再在网上找一个什么项目,凑两个,而不是用自己现在这两个项目?或者是点评改改放前面,然后原本第一个项目,把秒杀抽掉,剩下的想办法从网上火的RAG项目里移植点亮点,或者直接就用网上的RAG项目?感觉我主要还是偏向后端开发,但是感觉如果除开点评,再拿一个项目,想不到有什么自己能掌控且跟点评不重的。然后鼠鼠之前主要的问题是担心面试让打开项目演示,然后就一直花时间在用AI整第一个项目,第二个项目都没时间整,第四次面试之前还因为太害怕被认为不熟悉项目,跟AI一起把简历的说辞做了大幅度弱化,然后暑期都是拿弱化后的简历投的,感觉是不是看上去太没有吸引力就直接给简历挂了。(图1是弱化后的,图2是弱化前的,但之前3月初投了几家好像也是简历挂。)而且因为3月花了很多时间整在跟AI整代码,导致八股和算法都没怎么看,算法之前有跟灵神题单刷一些,还算入门,但是八股只看了一些基本的,可能面试的时候只答得上来60-70%,而且表述有些混乱,都是想到哪说到哪;前面几回面试基本上都有大板块的基础八股没答出来,比如RedisZ&nbsp;Set数据结构,MQ延时消息、可靠性保证,JVM内存分配的过程、GC&nbsp;roots,JUC锁,设计模式。现在有点不知道该怎么办。求大佬们给点简历修改建议或者面试准备建议,不胜感激!
何时能不做牛马:简历每个点之间的间距可以缩一下。几乎没遇到过要演示项目的情况,即使万一遇上了你也可以说部署在其他电脑上本地没代码。nku不应该简历挂吧?抓紧背背八股练练表达,不要放弃,五六月份找到也不晚(不然还得提前入职
应届生简历当中,HR最关...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务