小酷宝的二进制树

时间限制: 1 Sec  内存限制: 128 MB
提交: 0  解决: 12

题目描述

小酷宝拿到了一棵树，共有n个节点，每个节点都有一个权值: 0或者1。这样，每条路径就代表了一个二进制数。游游想知道，有多少条路径代表的二进制数在[ L , R ]区间范围内?(请注意: 路径长度至少为1，也就是包含2个节点，例如，节点3到节点3虽然有一个权值，但并不是合法路径)
例如n=4，l=4，r=5，节点权值为1010，  边为(1,2) (2,3) (3,4)  ，此时值在[ 4 , 5 ]之间的路径有3条
路径1-2-3代表的二进制数为5
路径3-2-1代表的二进制数为5
路径4-3-2-1代表的二进制数为5

输入

第一行输入三个正整数n,l,r，用空格隔开。
第二行输入一个长度为n的01串，第i个字符代表i号节点的权值
接下来的n-1行，每行2个正整数u和v，代表u号节点和v号节点有一条边连接。
1<=n<=1e3
1<=u,v<=n
1<=l<=r<=1e14

输出

一个整数，代表合法的路径条数

样例输入

4 4 5
1010
1 2
2 3
3 4

样例输出

3

代码：#include <bits/stdc++.h>
using namespace std;
int n;
long long l, r;
int ans = 0;
string s;
vector<int> g[1010];
void dfs(int u, int p, long long x)
{
    for (int v : g[u])     {
        if (v != p)
        {
            long long y = x * 2 + s[v] - '0';
            if (y <= r)
            {
                if(y>=l )ans ++;
                dfs(v, u, y);
            }
        }
    }
}
int main()
{
    cin >> n >> l >> r;
    cin >> s;
    s = " " + s;
    for (int i = 1, u, v; i < n; i += 1)
    {
        cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    
    for (int i = 1; i <= n; i += 1)
    {
        dfs(i, 0, s[i] - '0');
    }
    cout << ans;
}