HDU 3078 Network (lca+排序)

题意:一棵无向树,输入点数和操作数,下面一行n个值代表每个点的权。下面n-1行是树边。操作分为 0 x w ,表示把点x的权改为w; k a b , 求出从a到b的路径中,第k大的点权。

题解:lca+排序
居然直接暴力就可以过了,每次询问将lca路径中的点权排序即可。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
#define pii pair<int, int>
using namespace std;
const int MAXN = 8e4 + 5;
int n, q;
vector<int> v[MAXN];
int fa[MAXN][31], dep[MAXN];
void init() {
    for (int i = 1; i <= n; i++) v[i].clear();
    // memset(fa, 0, sizeof(fa));
    // memset(cost, 0, sizeof(cost));
    // memset(dep, 0, sizeof(dep));
}
void dfs(int root, int fno) {   //1 0
    fa[root][0] = fno;
    dep[root] = dep[fa[root][0]] + 1;
    for (int i = 1; i < 31; ++i) {
        fa[root][i] = fa[fa[root][i - 1]][i - 1];
    }
    int sz = v[root].size();
    for (int i = 0; i < sz; ++i) {
        if (v[root][i] == fno) continue;
        dfs(v[root][i], root);
    }
}
int lca(int x, int y) {
    if (dep[x] > dep[y]) swap(x, y);
    int tmp = dep[y] - dep[x];
    for (int j = 0; tmp; ++j, tmp >>= 1)
        if (tmp & 1) y = fa[y][j];
    if (y == x) return x;
    for (int j = 30; j >= 0 && y != x; --j) {
        if (fa[x][j] != fa[y][j]) {
            x = fa[x][j];
            y = fa[y][j];
        }
    }
    return fa[x][0];
}
int a[MAXN], k;
vector<int> vv;
int main() {
    while (~scanf("%d%d", &n, &q)) {
        init();
        int x, y;
        for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
        for (int i = 1; i < n; i++) {
            scanf("%d%d", &x, &y);
            v[x].push_back(y);
            v[y].push_back(x);
        }
        dfs(1, 0);
        while (q--) {
            scanf("%d%d%d", &k, &x, &y);
            if (k == 0) a[x] = y;
            else {
                int pa = lca(x, y);
                vv.clear();
                vv.push_back(a[pa]);
                while (x != pa) {
                    vv.push_back(a[x]);
                    x = fa[x][0];
                }
                while (y != pa) {
                    vv.push_back(a[y]);
                    y = fa[y][0];
                }
                if (vv.size() < k) puts("invalid request!");
                else {
                    sort(vv.begin(), vv.end());
                    printf("%d\n", vv[vv.size() - k]);
                }
            }
        }
    }
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值