bzoj4009

4009: [HNOI2015]接水果

Time Limit: 60 Sec  Memory Limit: 512 MB
Submit: 633  Solved: 299
[Submit][Status][Discuss]

Description

风见幽香非常喜欢玩一个叫做 osu!的游戏，其中她最喜欢玩的模式就是接水果。

由于她已经DT FC 了The big black,  她觉得这个游戏太简单了，于是发明了一个更

加难的版本。首先有一个地图，是一棵由 n 个顶点、n-1 条边组成的树（例如图 1

给出的树包含 8 个顶点、7 条边）。这颗树上有 P 个盘子，每个盘子实际上是一条

路径（例如图 1 中顶点 6 到顶点 8 的路径），并且每个盘子还有一个权值。第 i 个

盘子就是顶点a_i到顶点b_i的路径(由于是树，所以从a_i到b_i的路径是唯一的)，

权值为c_i。接下来依次会有Q个水果掉下来，每个水果本质上也是一条路径，第

i 个水果是从顶点 u_i 到顶点v_i 的路径。幽香每次需要选择一个盘子去接当前的水

果：一个盘子能接住一个水果，当且仅当盘子的路径是水果的路径的子路径（例如

图1中从 3到7 的路径是从1到8的路径的子路径）。这里规定:从a 到b的路径与

从b到 a的路径是同一条路径。当然为了提高难度，对于第 i 个水果，你需要选择

能接住它的所有盘子中，权值第 k_i 小的那个盘子，每个盘子可重复使用（没有使用次数

的上限：一个盘子接完一个水果后，后面还可继续接其他水果，只要它是水

果路径的子路径）。幽香认为这个游戏很难，你能轻松解决给她看吗？ 

 

Input

第一行三个数 n和P 和Q，表示树的大小和盘子的个数和水果的个数。 

接下来n-1 行，每行两个数 a、b，表示树上的a和b 之间有一条边。树中顶点

按1到 n标号。 接下来 P 行，每行三个数 a、b、c，表示路径为 a 到 b、权值为 c 的盘子，其

中0≤c≤10^9，a不等于b。 

接下来Q行，每行三个数 u、v、k，表示路径为 u到 v的水果，其中 u不等于v，你需要选择第 k小的盘子，

第k 小一定存在。 

 

Output

 对于每个果子，输出一行表示选择的盘子的权值。 

Sample Input

10 10 10
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
3 2 217394434
10 7 13022269
6 7 283254485
6 8 333042360
4 6 442139372
8 3 225045590
10 4 922205209
10 8 808296330
9 2 486331361
4 9 551176338
1 8 5
3 8 3
3 8 4
1 8 3
4 8 1
2 3 1
2 3 1
2 3 1
2 4 1
1 4 1

Sample Output

442139372
333042360
442139372
283254485
283254485
217394434
217394434
217394434
217394434
217394434

HINT

 

N,P,Q<=40000。 

 

Source

http://blog.csdn.net/thy_asdf/article/details/50363672

前面的部分说的很详细了。但是写的时候要注意几点：

1. 判断lca类型的时候u和v比较的是dfn大小，不是dep大小

2.水果读入的时候如果dfn[x1] > dfn[y1] swap(x1, y1)

3.整体二分里排序注意插入删除和水果的顺序

#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
int n, p, m, cnt, tot = 1, Time;
struct E {
    int nxt, to;
} e[N << 1];
struct P {
    int x1, y1, x2, y2, w, id, k, type;
    P(int x1 = 0, int y1 = 0, int x2 = 0, int y2 = 0, int w = 0, int id = 0, int k = 0, int type = 0) : x1(x1), y1(y1), x2(x2), y2(y2), w(w), id(id), k(k), type(type) {}
} event[N], q[N];
struct BIT {
    int tree[N];
    int lowbit(int x) {return x&(-x);}
    void update(int l, int r, int v) 
    {
        for(int i = l; i <= n; i += lowbit(i)) tree[i] += v;
        for(int i = r + 1; i <= n; i += lowbit(i)) tree[i] -= v;
    }
    int query(int pos) {int ret = 0; for(int i = pos; i; i -= lowbit(i)) ret += tree[i]; return ret;}
} t;
vector<P> c, lp, rp; 
int head[N], ans[N], low[N], dfn[N] ,fa[23][N], dep[N];
void link(int u, int v)
{
    e[++tot].nxt = head[u];
    head[u] = tot;
    e[tot].to = v;
}
bool cp(P i, P j) {return i.w < j.w;} 
bool cp1(P i, P j) {return i.x1 == j.x1 ? i.type < j.type : i.x1 < j.x1;}
void dfs(int u, int last)
{
    dfn[u] = ++Time;
    for(int i = head[u]; i; i = e[i].nxt) if(e[i].to != last)
    {
        fa[0][e[i].to] = u;
        dep[e[i].to] = dep[u] + 1;
        dfs(e[i].to, u);
    }
    low[u] = Time;
}
int lca(int u, int v)
{
    if(dep[u] < dep[v]) swap(u, v);
    for(int i = 22; i >= 0; --i) if((dep[u] - dep[v]) & (1 << i))
        u = fa[i][u];
    if(u == v) return u;
    for(int i = 22; i >= 0; --i) if(fa[i][u] != fa[i][v])
    {
        u = fa[i][u]; v = fa[i][v];
    }
    return fa[0][u];
}
int jump(int u, int h)
{
    for(int i = 22; i >= 0; --i) if(h & (1 << i)) u = fa[i][u]; return u;
}
void solve(int l, int r, int lb, int rb)
{
    if(l > r) return;
    if(lb == rb)
    {
        for(int i = l; i <= r; ++i) ans[q[i].id] = event[lb].w;
        return;
    }
    int mb = (lb + rb) >> 1;
    c.clear(); lp.clear(); rp.clear();
    for(int i = lb; i <= mb; ++i)
    {//盘子 
        P x = event[i]; 
//        printf("x1=%d x2=%d\n", x.x1, x.x2);
        x.w = 1; x.type = 0;
        c.push_back(x); x.type = 2;
        x.w = -1; x.x1 = event[i].x2;
        c.push_back(x);            
    }
    for(int i = l; i <= r; ++i)
    {//水果 
        c.push_back(q[i]);
    }
    sort(c.begin(), c.end(), cp1);
    for(int i = 0; i < c.size(); ++i)
    {
        if(c[i].type != 1) 
        {
//            printf("x=%d y1=%d y2=%d\n", c[i].x1, c[i].y1, c[i].y2);
            t.update(c[i].y1, c[i].y2, c[i].w);
        } 
        else 
        {
            int sum = t.query(c[i].y1);
//            printf("sum=%d k=%d\n", sum, c[i].k);
            if(sum >= c[i].k) lp.push_back(c[i]);
            else {c[i].k -= sum; rp.push_back(c[i]);}
        }
    }
    for(int i = 0; i < c.size(); ++i) if(c[i].type != 1)
        t.update(c[i].y1, c[i].y2, -c[i].w);
    for(int i = 0; i < lp.size(); ++i) q[l + i] = lp[i];
    for(int i = 0; i < rp.size(); ++i) q[l + lp.size() + i] = rp[i];
    int mid = l + lp.size() - 1;
    solve(l, mid, lb, mb); solve(mid + 1, r, mb + 1, rb);
}
int main()
{
    freopen("fruit_hnoi2015.in", "r", stdin);
    freopen("fruit_hnoi2015.out", "w", stdout);
    scanf("%d%d%d",&n,&p,&m);
    for(int i = 1; i < n; ++i)
    {
        int u, v; scanf("%d%d",&u,&v);
        link(u, v); link(v, u);
    }
    memset(fa, -1, sizeof(fa));
    dfs(1, 0);
    for(int i = 1; i <= 22; ++i)
        for(int j = 1; j <= n; ++j) if(fa[i-1][j] != -1)
            fa[i][j] = fa[i-1][fa[i-1][j]];
    for(int i = 1; i <= p; ++i)
    {
        int u, v, w; scanf("%d%d%d", &u, &v, &w);
        int x = lca(u, v);
        if(dfn[u] > dfn[v]) swap(u, v);
        if(u != x) 
        {
            event[++cnt].x1 = dfn[u]; event[cnt].y1 = dfn[v];
            event[cnt].x2 = low[u]; event[cnt].y2 = low[v]; 
            event[cnt].w = w; 
        }
        else
        {
            int t = jump(v, dep[v] - dep[u] - 1);
            event[++cnt].x1 = 1; event[cnt].y1 = dfn[v];
            event[cnt].x2 = dfn[t] - 1; event[cnt].y2 = low[v];
            event[cnt].w = w;
            if(low[t] < n)
            {
                event[++cnt].x1 = dfn[v]; event[cnt].y1 = low[t] + 1;
                event[cnt].x2 = low[v]; event[cnt].y2 = n;
                event[cnt].w = w; 
            }
        }
    }
    for(int i = 1; i <= m; ++i)
    {
        int u, v;
        scanf("%d%d%d", &u, &v, &q[i].k);
        if(dfn[u] > dfn[v]) swap(u, v);
        q[i].x1 = dfn[u]; q[i].y1 = dfn[v];
        q[i].id = i; q[i].type = 1;
    }
    sort(event + 1, event + cnt + 1, cp);
    solve(1, m, 1, cnt);
    for(int i = 1; i <= m; ++i) printf("%d\n", ans[i]);
    fclose(stdin); fclose(stdout);
    return 0;
}