Advertisement
MinhNGUYEN2k4

DELTREE

Feb 17th, 2022
758
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.44 KB | None | 0 0
  1. //Nguyen Huu Hoang Minh
  2. #include <bits/stdc++.h>
  3. #define sz(x) int(x.size())
  4. #define all(x) x.begin(),x.end()
  5. #define reset(x) memset(x, 0,sizeof(x))
  6. #define pb push_back
  7. #define mp make_pair
  8. #define fi first
  9. #define se second
  10. #define N 300005
  11. #define remain(x) if (x > MOD) x -= MOD
  12. #define ii pair<int, int>
  13. #define iiii pair< ii , ii >
  14. #define viiii vector< iiii >
  15. #define vi vector<int>
  16. #define vii vector< ii >
  17. #define bit(x, i) (((x) >> (i)) & 1)
  18. #define Task "deltree"
  19. #define int long long
  20.  
  21. using namespace std;
  22.  
  23. typedef long double ld;
  24. const int inf = 1e10;
  25. const int minf = -1e10;
  26.  
  27. int n;
  28. string s;
  29. vector<int> g[N];
  30. int dp[N][2];
  31.  
  32. void readfile()
  33. {
  34.     ios_base::sync_with_stdio(false);
  35.     cin.tie(0);cout.tie(0);
  36.     if (fopen(Task".inp","r"))
  37.     {
  38.         freopen(Task".inp","r",stdin);
  39.         freopen(Task".out","w",stdout);
  40.     }
  41.     cin >> n >> s;
  42.     s = " " + s;
  43.     for(int i=1; i<n; i++){
  44.         int u, v; cin >> u >> v;
  45.         g[u].pb(v);
  46.         g[v].pb(u);
  47.     }
  48. }
  49.  
  50. void dfs(int u, int par){
  51.     for(auto v : g[u]){
  52.         if (v==par) continue;
  53.         dfs(v,u);
  54.         dp[u][0] += min(dp[v][0],dp[v][1]+1);
  55.         dp[u][1] += min(dp[v][1],dp[v][0]+1);
  56.     }
  57.     dp[u][0] += ((s[u]-'0')==0 ? 0 : 1);
  58.     dp[u][1] += ((s[u]-'0')==1 ? 0 : 1);
  59. }
  60.  
  61. void proc()
  62. {
  63.     dfs(1,-1);
  64.     cout << min(dp[1][0],dp[1][1])+1;
  65. }
  66.  
  67. signed main()
  68. {
  69.     readfile();
  70.     proc();
  71.     return 0;
  72. }
  73.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement