題目描述:

給一棵樹,每條邊有權.求一條路徑,權值和等於K,且邊的數量最小.

輸入描述:

第一行 兩個整數 n, k 第二…n行 每行三個整數 表示一條無向邊的兩端和權值 (注意點的編號從0開始) (n≤200000,K≤1000000)

輸出描述:

一個整數 表示最小邊數量 如果不存在這樣的路徑 輸出-1

樣例輸入:

4 3
0 1 1
1 2 2
1 3 4

樣例輸出:

2

思路

利用點分治

開一個100W的陣列t,t[i]表示權值為i的路徑最少邊數 找到重心分成若干子樹後， 得出一棵子樹的所有點到根的權值和x，到根a條邊，用t[k-x]+a更新答案，全部查詢完後，再用所有a更新t[x]，這樣可以保證不出現點分治中的不合法情況

d[i]當前樹走到i點走過的邊數,deep[i]表示走過的路徑

程式碼

#include <bits/stdc++.h>
using namespace std;
#define mem(a, b) memset(a, b, sizeof(a))
const int inf = 0x3f3f3f3f;
const int N = 2e5 + 10;
const int M = 1e6 + 10;
int n, k;
int first[N], tot;
int f[N], siz[N], vis[N], ans, sum;
int d[N], deep[N]; //d[i]當前樹走到i點走過的邊數,deep[i]表示走過的路徑
 

int tmp[M], root;
struct edge
{
    int v, w, next;
} e[N * 2];
void add_edge(int u, int v, int w)
{
    e[tot].v = v, e[tot].w = w;
    e[tot].next = first[u];
    first[u] = tot++;
}
void init()
{
    mem(first, -1);
    tot = 0;
    root = 0;
    sum = f[0] = n;
    ans = inf;
}
void getroot(int u,
 
 int fa)
{
    siz[u] = 1, f[u] = 0;
    for (int i = first[u]; ~i; i = e[i].next)
    {
        int v = e[i].v;
        if (v == fa || vis[v])
            continue;
        getroot(v, u);
        siz[u] += siz[v];
        f[u] = max(f[u], siz[v]);
    }
    f[u] = max(f[u], sum - siz[u]);
    if (f[u] < f[root])
        root = u;
}
void getdeep(int u, int fa)
{
    if (deep[u] <= k)
        ans = min(ans, tmp[k - deep[u]] + d[u]);
    for (int i = first[u]; ~i; i = e[i].next)
    {
        int v = e[i].v, w = e[i].w;
        if (v == fa || vis[v])
            continue;
        deep[v] = deep[u] + w;
        d[v] = d[u] + 1;
        getdeep(v, u);
    }
}
void update(int u, int fa, int flag)
{
    if (deep[u] <= k)
    {
        if (flag)
            tmp[deep[u]] = min(tmp[deep[u]], d[u]);
        else
            tmp[deep[u]] = inf;
    }
    for (int i = first[u]; ~i; i = e[i].next)
    {
        int v = e[i].v;
        if (v == fa || vis[v])
            continue;
        update(v, u, flag);
    }
}
void solve(int u)
{
    vis[u] = 1;
    tmp[0] = 0;
    for (int i = first[u]; ~i; i = e[i].next)
    {
        int v = e[i].v, w = e[i].w;
        if (vis[v])
            continue;
        deep[v] = w;
        d[v] = 1;
        getdeep(v, 0);
        update(v, 0, 1);
    }
    for (int i = first[u]; ~i; i = e[i].next)
    {
        int v = e[i].v;
        if (!vis[v])
            update(v, 0, 0);
    }
    for (int i = first[u]; ~i; i = e[i].next)
    {
        int v = e[i].v;
        if (vis[v])
            continue;
        sum = siz[v];
        root = 0;
        getroot(v, 0);
        solve(root);
    }
}
int main()
{
    int u, v, w;
    scanf("%d%d", &n, &k);
    init();
    for (int i = 1; i <= k; i++)
        tmp[i] = inf;
    for (int i = 1; i <= n - 1; i++)
    {
        scanf("%d%d%d", &u, &v, &w);
        u++, v++;
        add_edge(u, v, w);
        add_edge(v, u, w);
    }
    getroot(1, 0);
    solve(root);
    if (ans != inf)
        printf("%d\n", ans);
    else
        printf("-1");

    return 0;
}