1741 Tree (树的分治,点分治)

原创 2013年12月03日 18:19:05

参考:

http://hi.baidu.com/strongoier/item/fe47a4191c18a37c1009b515

http://hi.baidu.com/shuxk/item/2bd327977576038159146161

http://hi.baidu.com/gugugupan/item/f29befcd12accc67f7c95db9

给出N(1 <= N <= 10000)个结点的树,求使得路径u -> v长度不超过k的点对(u, v)的个数。

点分治法:每次找出分治点,求经过其的路径的点对,子树递归处理即可。

会有重复值,需要减去子节点相关的一些值

注意技巧:

删除点的办法。adj[r ^ 1].mk = false;

找出分支点的方法。(与直径中心的差别?)

算法就是基于树的分治,对于每个树,先要选定一个根节点,这个根节点要保证它的深度尽量小,然后才可以保证时间复杂度。选根的过程大致就是一次DFS,找到以 I 为根节点时儿子最多的子树的儿子树最少的 I 就可以作为根了。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <ctime>
#include <cstdlib>
#include <cstring>
#include <queue>
#include <string>
#include <set>
#include <stack>
#include <map>
#include <cmath>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;

#define FE(i, a, b) for(int i = (a); i <= (b); ++i)
#define FD(i, b, a) for(int i = (b); i >= (a); --i)
#define REP(i, N) for(int i = 0; i < (N); ++i)
#define CLR(a, v) memset(a, v, sizeof(a))
#define PB push_back
#define MP make_pair

#define RI(n) scanf("%d", &n)
#define RIII(a, b, c) scanf("%d%d%d",&a, &b, &c)

typedef long long LL;
const double eps = 1e-10;
const int maxn = 10010;

int a[maxn], an;
int sz[maxn], szn;
int f[maxn], d[maxn];
int n, m;
int cur;///curroot
int ans;

struct Edge {
    int to, w, next;
    bool mk;
}adj[maxn * 2];
int adj_tot, head[maxn];
void adj_add(int x, int y, int w)
{
    adj[adj_tot].to = y; adj[adj_tot].w = w;
    adj[adj_tot].next = head[x];
    adj[adj_tot].mk = true;
    head[x] = adj_tot++;
}

void find_root(int u, int fa)///找分治点
{
    sz[u] = 1;///
    f[u] = 0;
    for (int r = head[u]; ~r; r = adj[r].next)
    {
        int v = adj[r].to;
        if (adj[r].mk && v != fa)
        {
            find_root(v, u);
            sz[u] += sz[v];
            f[u] = max(f[u], sz[v]);
        }
    }
    f[u] = max(f[u], szn - sz[u]);
    if (f[u] < f[cur]) cur = u;
}

void dfs(int u, int fa)
{
    sz[u] = 1;///
    a[an++] = d[u];
    for (int r = head[u]; ~r; r = adj[r].next)
    {
        int v = adj[r].to;
        if (adj[r].mk && v != fa)
        {
            d[v] = d[u] + adj[r].w;
            dfs(v, u);
            sz[u] += sz[v];
        }
    }
}

int calc(int x, int initval)
{
    int ret = an = 0;
    d[x] = initval;
    dfs(x, 0);
    sort(a, a + an);
    int l = 0, r = an - 1;
    for (; l < r;)///!!!
    {
        if (a[r] + a[l] <= m)
            ret += r - l++;
        else r--;
    }
    return ret;
}

void solve(int u)
{
    ans += calc(u, 0);
    for (int r = head[u]; ~r; r = adj[r].next)
    {
        int v = adj[r].to;
        if (adj[r].mk)
        {
            adj[r ^ 1].mk = false;///!!!
            ans -= calc(v, adj[r].w);///???

            f[0] = szn = sz[v];
            find_root(v, cur = 0);
            solve(cur);
        }
    }
}

int main ()
{
    int x, y, w;
    while (scanf("%d%d", &n, &m) != EOF && n + m )
    {
        adj_tot = 0; CLR(head, -1);
        ans = 0;
        for (int i = 0; i < n - 1; i++)
        {
            scanf("%d%d%d", &x, &y, &w);
            adj_add(x, y, w); adj_add(y, x, w);
        }
        f[0] = szn = n;
        find_root(1, cur = 0);
        solve(cur);
        printf("%d\n", ans);
    }
    return 0;
}


版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

POJ 1741 Tree DP+树的点分治

POJ 1741 Tree DP+树的点分治

POJ 1741 Tree (男人八题 - 树分治[点分])

题意: 树上n 个点, 求有多少点对满足距离不超过K 思路: 树上点分治 入门题目。 我们每次递归 都要从这棵树的重心开始递归。  从这个点这出其他点到这个点距离来, 然后我们看看有几个点对 ...

POJ 1741 Tree + POJ 1987 Distance Statistics【树的点分治】

Tree Time Limit: 1000MS Memory Limit: 30000K Total Submissions: 13558 Accepted: 4367 ...

【树的点分治】【平衡树】[POJ1741]Tree

题目大意给出一棵树,边带权,问有多少条长度路径<=k的路径。分析点分治暴力做法为O(n2)O(n^2)我们可以将统计所有路径转化为递归查找通过某个特殊点的路径,从而降低复杂度。将这个点定为重心,复杂度...

POJ 1741 Tree (树分治之点分治)

题目大意: 给出一棵树, 点数n 大致思路: 马上就要区域赛了...走前看点点分治的题...遵循昊哥的教导 这个题就是点分治很裸的一个模型了 所谓点分治, 每次将问题递归到子树上解...

POJ 1741 Tree, 树的重心, 树分治, 点分治

最近在学习树的分治,算是比较难,而且代码量比较大的一块。随便拿一道题来就有上百行,故写一篇文章来总结一下这方面的框架。 POJ这一题应该算是树分治的入门题,顺便用这一题来详细说明树分治的一些具体内容...

POJ 1741 Tree [树上点分治]

给出一棵树,问树上点对所代表的简单路径长度<=K的不同点对数。

POJ-1741 TREE 点分治 树上问题

大家都很强,可与之共勉函数对应关系写错, 调了两个小时, FUKK。 大致思路:点分支。 找以每一个节点(其实不是每一个)为根节点的重心, 用dis[ ]表示经过重心的每一条链的长度。但是计算的时...

poj1741 tree 点分治

poj1741 tree 把这几天写的东西上传一下 写博客的意义大概就是整理一下思路 以及方便日后复习 不得不说 把代码放在文件夹里 做完我是不会过目的 写博客 或许会增加我的成就感 从而...

poj1741 Tree 点分治

第一次写点分治啊,果然还是黄学长的代码框架好。。。     这绝对是点分治的最经典的题目了。大概写一下自己的理解吧。     首先,对于一棵树,求出其重心并作为根节点。然后链就可以分为两类:经过根节点...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

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