小红的二进制数
题目大意
一个01串从第i个位置出发到第j个位置结尾且第j个位置为1的路径条数,只能从父亲走到儿子
输入
1.n(1<=n<=1e5)
2.长度为n的01串
3.n-1条边
分析:树形dp或者说dfs,O(n)的复杂度
#include<bits/stdc++.h>
using namespace std;
int n;
string s;
int dp[100100];
vector<int>g[100100];
void dfs(int x,int fa){
dp[x] = s[x]-'0';
for(auto i :g[x]){
if(i==fa) continue;
dfs(i,x);
dp[x]+=dp[i];
}
}
int main(){
cin>>n>>s;
s='0'+s;
for(int i = 1;i<n;++i){
int x,y;
cin>>x>>y;
g[x].push_back(y);
g[y].push_back(x);
}
dfs(1,0);
for(int i = 1;i<=n;++i) cout<<dp[i]-(s[i]-'0')<<"\n";
return 0;
}