码迷,mamicode.com
首页 > 其他好文 > 详细

Codeforces 1304E. 1-Trees and Queries

时间:2020-02-16 14:55:36      阅读:74      评论:0      收藏:0      [点我收藏+]

标签:get   dfs   node   之间   快速   floor   ORC   isp   如何   

简述题意,给你一课最小支撑树,对每个询问,在原有的路径上增加x-y,问a-b是否有路径长度为k的路,每条路每个点可以重复使用

由于是最小支撑树,我们可以用LCA来快速判断每个点之间的距离,那么现在就要判断情况,假设从原有的路上,a-b的距离为d,d=k时显然成立,当d<k时,若(d-k)%2=0也成立,因为若其是2的倍数,他可以轮流进入b与b的前驱,最后停在b上,那么我们如何判断新加的边的,距离的判定和前者一样,但是距离的大小变化了,设(x,y)表示x到y的最短路径,则从a到b就有三种情况

1.(a,b) 该情况已经讨论过

2.(a,x)+(b,y)+1,意思从a到x,x到y的路径长度为1,y再到b

3.(a,y)+(b,x)+1,同理

三种情况下的距离判断都一样,具体上代码看一下就懂了

技术图片
#include<bits/stdc++.h>
using namespace std;
#define lowbit(x) ((x)&(-x))
typedef long long LL;

const int maxm = 1e5+5;
struct Node {
    int v, nex;
} edges[maxm<<1];

int head[maxm<<1], edgecnt, grand[maxm][21], depth[maxm], limit;

void addedge(int u, int v) {
    edges[++edgecnt] = {v, head[u]};
    head[u] = edgecnt;
}

void dfs(int u, int fa) {
    depth[u] = depth[fa] + 1;
    grand[u][0] = fa;
    for(int i = 1; i <= limit; ++i) grand[u][i] = grand[grand[u][i-1]][i-1];
    for(int i = head[u]; i; i = edges[i].nex) {
        int v = edges[i].v;
        if(v != fa) dfs(v, u);
    }
}

int lca(int a, int b) {
    if(a == b) return a;
    if(depth[a] > depth[b]) swap(a, b);
    for(int i = limit; i >= 0; i--) 
        if(depth[a] <= depth[b] - (1<<i)) b = grand[b][i];
    if(a == b) return a;
    for(int i = limit; i >= 0; --i) {
        if(grand[a][i] == grand[b][i])
            continue;
        else {
            a = grand[a][i], b = grand[b][i];
        }
    }
    return grand[a][0];
}

int getdist(int a, int b) {
    int c = lca(a, b);
    return depth[a] + depth[b] - 2*depth[c];
}


void run_case() {
    int n; cin >> n;
    limit = floor(log(n+0.0) / log(2.0)) + 1;
    int u, v;
    for(int i = 0; i < n-1; ++i) {
        cin >> u >> v;
        addedge(u, v), addedge(v, u);
    }
    dfs(1, 0);
    int q; cin >> q;
    int a, b, x, y, k;
    while(q--) {
        cin >> x >> y >> a >> b >> k;
        int dist = getdist(a, b);
        bool flag = false;
        if(dist <= k && (k-dist)%2==0) flag = true;
        dist = getdist(a, x) + getdist(b, y) + 1;
        if(dist <= k && (k-dist)%2==0) flag = true;
        dist = getdist(a, y) + getdist(b, x) + 1;
        if(dist <= k && (k-dist)%2==0) flag = true;
        if(flag) cout << "YES\n";
        else cout << "NO\n";
    }
}
 
int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    //cout.setf(ios_base::showpoint);cout.precision(10);
    //int t; cin >> t;
    //while(t--)
    run_case();
    cout.flush();
    return 0;
}
View Code

 

Codeforces 1304E. 1-Trees and Queries

标签:get   dfs   node   之间   快速   floor   ORC   isp   如何   

原文地址:https://www.cnblogs.com/GRedComeT/p/12316678.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!