P2286 [HNOI2004]宠物收养场

第一次用set水splay好爽啊！

---

这道题很明显，就是弄两个平衡树（一个也可以）。

进来一个人，就去另一边找找前驱和后继，优先找前驱，找个绝对值最小的配对走。

其实用一下set就可以了啊！

使用std::set.lower_bound就可以找到x元素，--一下就得到前驱，++一下应该就得到了后继。

其实后继也直接可用upper_bound。

再根据题意模拟一下就能过了。。。

这道题数据特别水。。。

代码：

#include<cstdio>
#include<algorithm>
const int maxn = 1005;
struct Edges
{
    int next, to, weight;
} e[maxn];
int head[maxn], tot;
int dp[maxn][maxn];
int n, m;
void link(int u, int v, int w)
{
    e[++tot] = (Edges){head[u], v, w};
    head[u] = tot;
}
int dfs(int u)
{
    int ans = 0;
    for(int i = head[u]; i; i = e[i].next)
    {
        int v = e[i].to;
        int t = dfs(v);
        ans = ans + t + 1;
        for(int j = std::min(ans, m); j > 0; j--)
        {
            for(int k = std::min(j - 1, ans); k > 0; k--)
            {
                dp[u][j] = std::max(dp[u][j], dp[u][j - k - 1] + dp[v][k] + e[i].weight);
            }
        }
    }
    return ans;
}
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i < n; i++)
    {
        int u, v, w;
        scanf("%d%d%d", &u, &v, &w);
        link(u, v, w);
    }
    dfs(1);
    printf("%d\n", dp[1][m]);
    return 0;
}