【uoj#150】【NOIP2015】运输计划 树上前缀和+lca+二分+拓扑排序+特别的卡常数技巧

公元 2044 年,人类进入了宇宙纪元。

L 国有 nn 个星球,还有 n−1n−1 条双向航道,每条航道建立在两个星球之间,这 n−1n−1 条航道连通了 LL 国的所有星球。

小 P 掌管一家物流公司, 该公司有很多个运输计划,每个运输计划形如:有一艘物流飞船需要从 uiui 号星球沿最快的宇航路径飞行到 vivi 号星球去。显然,飞船驶过一条航道是需要时间的,对于航道 jj,任意飞船驶过它所花费的时间为 tjtj,并且任意两艘飞船之间不会产生任何干扰。

为了鼓励科技创新, L 国国王同意小 P 的物流公司参与 L 国的航道建设,即允许小P 把某一条航道改造成虫洞,飞船驶过虫洞不消耗时间。

在虫洞的建设完成前小 P 的物流公司就预接了 mm 个运输计划。在虫洞建设完成后,这 mm 个运输计划会同时开始,所有飞船一起出发。当这 mm 个运输计划都完成时,小 P 的物流公司的阶段性工作就完成了。

如果小 P 可以自由选择将哪一条航道改造成虫洞, 试求出小 P 的物流公司完成阶段性工作所需要的最短时间是多少?

输入格式

第一行包括两个正整数 n,mn,m,表示 L 国中星球的数量及小 P 公司预接的运输计划的数量,星球从 11 到 nn 编号。

接下来 n−1n−1 行描述航道的建设情况,其中第 ii 行包含三个整数 ai,biai,bi 和 titi,表示第 ii 条双向航道修建在 aiai 与 bibi 两个星球之间,任意飞船驶过它所花费的时间为 titi。数据保证 1≤ai,bi≤n1≤ai,bi≤n 且 0≤ti≤10000≤ti≤1000。

接下来 mm 行描述运输计划的情况,其中第 jj 行包含两个正整数 ujuj 和 vjvj,表示第 jj 个运输计划是从 ujuj 号星球飞往 vjvj号星球。数据保证 1≤ui,vi≤n1≤ui,vi≤n

输出格式

输出文件只包含一个整数,表示小 P 的物流公司完成阶段性工作所需要的最短时间。

样例一

input

6 3
1 2 3
1 6 4
3 1 7
4 3 6
3 5 5
3 6
2 5
4 5

output

11

explanation

将第 11 条航道改造成虫洞: 则三个计划耗时分别为:11,12,1111,12,11,故需要花费的时间为 1212。

将第 22 条航道改造成虫洞: 则三个计划耗时分别为:7,15,117,15,11,故需要花费的时间为 1515。

将第 33 条航道改造成虫洞: 则三个计划耗时分别为:4,8,114,8,11,故需要花费的时间为 1111。

将第 44 条航道改造成虫洞: 则三个计划耗时分别为:11,15,511,15,5,故需要花费的时间为 1515。

将第 55 条航道改造成虫洞: 则三个计划耗时分别为:11,10,611,10,6,故需要花费的时间为 1111。

故将第 33 条或第 55 条航道改造成虫洞均可使得完成阶段性工作的耗时最短,需要花费的时间为 1111。

限制与约定

【uoj#150】【NOIP2015】运输计划 树上前缀和+lca+二分+拓扑排序+特别的卡常数技巧_第1张图片
时间限制:1s1s
空间限制:256MB

UOJ某题解

因为要求最大时间最小,容易想到二分,

二分时间mid,我们需要修改的路径是这个路径的时间大于mid的,看看能否修改到小于等于mid。于是现在问题变成了多个路径求交,然后看看它们的交集中把某个边权变为0能否让所有路径的时间小于mid。

然后有神犇想到了树上前缀和(Orz),做法是 sum[x]++,sum[y]++,sum[lca(x,y)]=2 ,很容易想到这是想干什么…

然后就O(n)枚举了…

复杂度 O(nlogMAXN) ,还有大常数…

我的卡常数技巧:把每次求前缀和的dfs过程改为对一个拓扑排序好了的数组求和,手动读入,memset改成for到n清零,提前按时间排序然后二分的时候少for点…

话说这题总体来说不算难啊为什么我傻X15分

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;

const int INF = 300000000;
const int SZ = 600010;

int n,m;

int head[SZ],nxt[SZ],tot = 1;

struct edge{
    int t,d;
}l[SZ];

void build(int f,int t,int d)
{
    l[++ tot].t = t;
    l[tot].d = d;
    nxt[tot] = head[f];
    head[f] = tot;
}

int anc[SZ][30],dist[SZ],deep[SZ],cost[SZ];


void dfs(int u,int fa,int d)
{
    dist[u] = d;
    deep[u] = deep[fa] + 1;
    anc[u][0] = fa;
    for(int i = 1;anc[u][i - 1];i ++)
    {
        anc[u][i] = anc[anc[u][i - 1]][i - 1];
    }
    for(int i = head[u];i;i = nxt[i])
    {
        int v = l[i].t;
        if(v == fa) continue;
        cost[v] = l[i].d;
        dfs(v,u,d + l[i].d);
    }
}

int ask_lca(int x,int y)
{
    if(deep[x] < deep[y]) swap(x,y);
    if(deep[x] > deep[y])
    {
        int dd = deep[x] - deep[y];
        for(int i = 19;i >= 0;i --)
            if(dd & (1 << i))
                x = anc[x][i];
    }
    if(x != y)
    {
        for(int i = 19;i >= 0;i --)
            if(anc[x][i] != anc[y][i])
                x = anc[x][i],y = anc[y][i];
    }
    if(x == y) return x;
    else return anc[x][0];
}

struct haha{
    int x,y,lca,d;
}ask[SZ];

int tm[SZ];

int S[SZ],top = 0;

bool check(int mid)
{

    for(int i = 1;i <= n;i ++) tm[i] = 0;

    int tot = 0,maxn = 0;
    for(int i = 1;i <= m;i ++)
    {
        if(ask[i].d > mid)
        {
            tm[ask[i].x] ++;
            tm[ask[i].y] ++;
            tm[ask[i].lca] -= 2;
            tot ++;
            maxn = max(maxn,ask[i].d);
        }
        else break;
    }

    for(int i = top;i >= 0;i --)
        tm[anc[S[i]][0]] += tm[S[i]];

    for(int i = 1;i <= n;i ++)
        if(tm[i] == tot)
            if(maxn - cost[i] <= mid) return true;
    return false;
}

int div()
{
    int l = -1,r = INF;
    while(r - l > 1)
    {
        int mid = (l + r) >> 1;
        if(check(mid)) r = mid;
        else l = mid;
    }
    return r;
}

queue<int> q;

void bfs()
{
    q.push(1);
    while(q.size())
    {
        int u = q.front(); q.pop();
        S[++ top] = u;
        for(int i = head[u];i;i = nxt[i])
            if(l[i].t != anc[u][0])
                q.push(l[i].t);
    }
}


void scanf(int &x)
{
    x = 0;
    char a = getchar();
    bool flag = 0;
    while(a > '9' || a < '0') {if(a == '-') flag = 1; a = getchar();}
    while(a >= '0' && a <= '9') x = x * 10 + a - '0',a = getchar();
    if(flag) x = -x;
}

bool cmp(haha a ,haha b)
{
    return a.d > b.d;
}

int main()
{
    scanf(n);scanf(m);
    for(int i = 1;i < n;i ++)
    {
        int a,b,c;
        scanf(a);
        scanf(b);
        scanf(c);
        build(a,b,c); build(b,a,c);
    }

    dfs(1,0,0);

    for(int i = 1;i <= m;i ++)
    {
        scanf(ask[i].x);
        scanf(ask[i].y);
        ask[i].lca = ask_lca(ask[i].x,ask[i].y);
        ask[i].d = dist[ask[i].x] + dist[ask[i].y] - 2 * dist[ask[i].lca];
    }
    sort(ask + 1,ask + 1 + m,cmp);
    bfs();
    printf("%d",div());
    return 0;
}

你可能感兴趣的:(【uoj#150】【NOIP2015】运输计划 树上前缀和+lca+二分+拓扑排序+特别的卡常数技巧)