【bzoj2783】【JLOI2012】树

最新推荐文章于 2021-11-07 11:35:00 发布

原创最新推荐文章于 2021-11-07 11:35:00 发布 · 304 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #bzoj #省选

题解同时被 2 个专栏收录

56 篇文章

订阅专栏

搜索

6 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索(DFS)解决特定树形结构中路径计数问题的方法。给定一棵包含n个节点的树及各节点权重，通过DFS算法计算从任意节点出发向下遍历至其它节点时，路径总权重等于指定值S的路径数量。

题意

有一棵 $n$ 个节点的树，每个节点有一个权值 $W_i$
一条路径的权值为 $\sum{W_u}$ ， $u$ 为路径上的点（包括端点）
求权值为 $S$ 的，从上往下的路径条数

解法

$DFS$ ：
从每一个节点往下进行 $DFS$ ，记录累计权值 $sum$ ，当 $sum≥S$ 的时候就返回，如果 $sum=S$ ，就对答案产生贡献

复杂度

O（ $nlen$ ）， $len$ 为所求路径的平均长度，一般在 $【10，20】之间$

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
using namespace std;
const int MAXN=100010;
struct node
{
    int next,to;
}t[MAXN];
int head[MAXN],num;
int w[MAXN];
int n,s,ans;
void add(int u,int v)
{
    t[++num]=(node){ head[u],v };
    head[u]=num;
}
void dfs(int k,int sum)
{
    if( sum>=s )   { ans+=(sum==s);return ; }
    for(int i=head[k],x; i ;i=t[i].next)
    {
        x=t[i].to;
        dfs( x,sum+w[x] );
    }
}
int main()
{
    int u,v;
    scanf("%d%d",&n,&s);
    for(int i=1;i<=n;i++)   scanf("%d",&w[i]);
    for(int i=1;i<=n-1;i++)
    {
        scanf("%d%d",&u,&v);
        add( u,v );
    }
    for(int i=1;i<=n;i++)   dfs( i,w[i] );
    printf("%d\n",ans);
    return 0;
}