hdu 5664 Lady CA and the graph (树分治,树状数组)

原创 2016年05月30日 20:50:37

http://acm.hdu.edu.cn/showproblem.php?pid=5664


题意:定义folded chain为一条路径u,v, u v的lca != u && != v, 求第k大folded chain


#include <stdio.h>
#include <bits/stdc++.h>
using namespace std;


#pragma comment(linker, "/STACK:102400000,102400000")
#define eps 1e-9
#define LL long long 
#define ULL unsigned long long
#define pii pair<int,int>
#define MP make_pair
#define N (100000 + 10)
#define M (200000 + 10)
#define inf 0x3f3f3f3f
#define mod 1000000007
#define lson ls, ll, md
#define rson rs, md+1, rr


int read() {
    char c;
    while((c=getchar()) && !('0' <= c && c <= '9'));
    int ret = c -'0';
    while((c=getchar()) && '0' <=  c && c <= '9') ret = ret * 10 + c - '0';
    return ret;
}

int fst[N], vv[M], nxt[M], cost[M], e;

void init() {
    memset(fst, -1, sizeof fst);
    e = 0;

}

void add(int u, int v, int w) {
    vv[e] = v, cost[e] = w, nxt[e] = fst[u], fst[u] = e++;
}

int root, mx[N], sz[N], tot;
bool vis[N];
vector<int> que[N*20], all[N];
vector<int> lab[N];
int cnt;

void dfs1(int u, int p) {
    sz[u] = 1;
    for(int i = fst[u]; ~i; i= nxt[i]) {
        int v = vv[i];
        if(v == p || vis[v]) continue;
        dfs1(v, u);
        sz[u] += sz[v];
    }
}

void dfs2(int u, int p) {
    mx[u] = 0;
    for(int i = fst[u]; ~i; i = nxt[i] ){
        int v = vv[i];
        if(v == p || vis[v]) continue;
        dfs2(v, u);
        mx[u] = max(mx[u], sz[v]);
    }
    mx[u] = max(tot-sz[u], mx[u]);
    if(root == -1 || mx[u] < mx[root]) {
        root = u;
    }
}


void find_root(int &u) {
    dfs1(u, -1);
    tot = sz[u];
    root = -1;
    dfs2(u, -1);
    u = root;
}

void dfs(int u, int p, int dis) {
    all[root].push_back(dis);
    que[cnt].push_back(dis);
    for(int i = fst[u]; ~i; i = nxt[i]) {
        int v = vv[i];
        if(v == p || vis[v]) continue;
        dfs(v, u, dis+cost[i]);
    }
}

int mid;
LL ans;
int n,m;
LL k;
LL cal(vector<int> &que, int k) {
    if(k == 0) sort(que.begin(), que.end());
    else {
        LL ans = 0;
        int j = 0;
        for(int i = que.size()-1; i >= 0; --i) {
            while(j < que.size() && que[j] + que[i] < mid) ++j;
            ans += max(0, i-j);
        }
        return ans;
    }
    return 0;
}


void solve(int u) {
    find_root(u);
    vis[u] = 1;
    all[u].clear();
    lab[u].clear();
    all[u].push_back(0);
    for(int i = fst[u]; ~i; i = nxt[i] ){
        int v = vv[i];
        if(vis[v]) continue;
        ++cnt;
        que[cnt].clear();
        lab[u].push_back(cnt);
        dfs(v, u, cost[i]);
        cal(que[cnt], 0);
    }
    cal(all[u], 0);

    for(int i = fst[u]; ~i; i= nxt[i]) {
        int v = vv[i];
        if(vis[v]) continue;
        solve(v);
    }
}


int san[N<<2], scnt;
int sum[N];
void f(int u, int p, int dis) {
    san[++scnt] = dis;
    san[++scnt] = dis-mid;
    for(int i = fst[u]; ~i; i = nxt[i]) {
        int v = vv[i];
        if(v == p) continue;
        f(v, u, dis+cost[i]);
    }
}


void add(int x, int v) {
    while(x <= scnt) {
        sum[x] += v;
        x += x & -x;
    }
}

int query(int x) {
    int s = 0;
    while(x) {
        s += sum[x];
        x -= x & -x;
    }
    return s;
}

int hx(int x) {
    return lower_bound(san+1, san+scnt+1, x) - san;
}

void g(int u, int p, int dis){ 
    int id = hx(dis);
    ans -= query(hx(dis-mid));
    add(id, 1);
    for(int i = fst[u]; ~i; i = nxt[i]) {
        int v = vv[i];
        if(v == p) continue;
        g(v, u, dis+cost[i]);
    }
    add(id, -1);
}

int gao() {
    for(int i = 1; i <= n;++i)
        ans += cal(all[i], 1);
    for(int i = 1; i <= cnt; ++i)
        ans -= cal(que[i], 1);
}

bool check() {
    ans = 0;
    for(int i = 1; i <= n; ++i) vis[i] = 0;
    gao();
    scnt = 0;
    f(m, -1, 0);
    sort(san, san+scnt);
    scnt = unique(san, san+scnt) - san;
    g(m, -1, 0);
    return ans >= k;
}

int main() {
    int T;
    T = read();
    while(T--) {
        n = read(); 
        m = read();
        k = read();
        init();
        int Max = 0;
        for(int i = 1; i < n; ++i) {
            int u, v, w;
            u = read(); v = read(); w = read();
            Max = max(Max, w);
            add(u,v, w);
            add(v, u, w);
        }

        memset(vis, 0, sizeof vis);
        cnt = 0;
        solve(m);

        int l = 0, r = n*Max;;
        int ret = 0;
        while(l < r) {
            mid = l + r >> 1;
            if(check()) l = mid+1, ret = mid;
            else r = mid;
        }
        if(ret == 0) puts("NO");
        else
        printf("%d\n", ret);
    }
}

Hdu 5664 Lady CA and the graph(有n个点的树,给定根,叫你找第k大的特殊链)

传送门:Hdu 5664 Lady CA and the graph 题意: 给你一个有n个点的树,给定根,叫你找第k大的特殊链 特殊的链的定义:u,v之间的路径,且lca(u,v)!=...

HDU 5333 Undirected Graph 离线 LCT维护最大生成树+树状数组

传送门【HDU 5333】 题意:一个无向图,m条边,Q个询问,每个询问查询若只保留图中的连接的两个点都在L,R之间的边,图包含多少个联通块。 思路:对询问以及边按照上届R为第一关键字从小到大排序...

hdu 5412 CRB and Queries(动态区间第k大值,区间能修改)(整体二分,树状数组套平衡树,线段树套treap)

题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=5412 解题思路: 官方题解: In this problem, we can u...

hdu 5338 ZZX and Permutations 2015多校联合训练赛,贪心,线段树,树状数组

ZZX and Permutations Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 131072/131072 K (Java/O...

多校第四场 1012 hdu 5338 ZZX and Permutations(线段树+树状数组+贪心)

题目链接:hdu 5338题目大意: 首先我们有一个序列:1,2,…….,n。 我们要读入一个操作序列,这个序列是1~n的一个排列,我们可以通过加括号形成一个循环节,循环节不能嵌套和交叉 循环节内部是...

HDU1166 敌兵布阵 线段树||树状数组||CDQ分治 入门题复习

敌兵布阵Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others) Total Submis...

HDU 5057 Argestes and Sequence(树状数组)

题目描述: 给你N个数,有M个查询,每次查询输入一个字符“Q”、“S”,为“Q”时输入四个数,“L”、“R”、“D”、“P”,代表查询这N个数从“L"到”R“之间第D位(从个位数起)为”P“的个数。...
  • stiyet
  • stiyet
  • 2016年07月18日 15:54
  • 220

hdu 5057 Argestes and Sequence (离散的树状数组)

Argestes and Sequence Time Limit: 5000/2500 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/O...

【HDU5654 BestCoder Round 77 (div1) D】【前驱位置思想 排序 树状数组】xiaoxin and his watermelon candy 区间内多少个不同连续单升三元

xiaoxin and his watermelon candy  Accepts: 15    Submissions: 172  Time Limit: 4000/4000 MS (J...

hdu 4417 Super Mario(主席树||cdq分治+树状数组)

主席树和cdq分治都可以,而且都是很入门的做法。 这道题主席树的写法要好写一些,但是如果带修改的话,就得用cdq分治了。 主席树代码: #include using namespac...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:hdu 5664 Lady CA and the graph (树分治,树状数组)
举报原因:
原因补充:

(最多只允许输入30个字)