fork download 
  1. // ⣿⣿⣿⣿⣿⠃⢿⣷⠆⠀⠀⠀⠀⠀⠀⠀⢸⣿⣿⣿⣿⣿⣯⡀⠀⠳⠤⠒⢦⡀⢠⣠⡿⠀⢸⣿⣿⡟⢳⡄⠈⠉⠹⣷⣄⠀⠀⠀⠀⠀
  2. // ⣿⣿⣿⣿⡏⠀⠈⢻⣦⣀⡀⠀⠀⠀⠀⠀⢸⣿⣿⣿⣿⣿⣿⣷⣄⡀⠀⠀⣾⠳⠞⠀⣤⣴⣿⣿⡟⠀⢀⣹⡄⠀⠀⠘⣿⣦⡀⠀⠀⠀
  3. // ⣿⣿⣿⣿⡁⠀⠀⠀⠉⢉⣙⡻⢶⣶⠆⠀⢸⣿⣿⣿⣿⣿⣿⣿⣿⣿⣶⣾⣿⣷⣄⣼⣿⣿⣿⠏⢠⣏⠻⣯⡙⠦⣀⠀⠈⢿⣿⣦⣄⠀
  4. // ⣿⣿⣿⣿⣿⣦⡀⠀⠀⣿⡿⢷⢰⣿⡇⠀⠀⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⠋⠀⠸⣌⠳⣌⠙⠲⣬⠙⢦⠘⣿⣿⣿⣷
  5. // ⣿⣿⣿⣿⣿⣿⣿⣦⡠⠿⣿⣷⣼⣿⠁⠀⠀⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡟⠁⢰⡦⣄⡘⢆⠀⠁⢀⠘⢧⠘⣇⠘⣿⣿⣿
  6. // ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣦⡀⠈⠛⠿⣄⣀⣤⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡿⢧⡀⠀⠀⠙⢦⡈⠙⢧⡀⠀⠇⠈⠁⠙⢧⣭⠻⣿
  7. // ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣶⣶⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⠿⠋⠀⠈⢳⠀⠀⠀⠀⠙⢦⠀⠁⠀⠀⠀⠀⠀⠀⢻⡀⠈
  8. // ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⠟⠁⠀⠀⠀⠀⠀⢧⠀⠀⠀⠀⠘⣇⠀⠀⠀⠀⠀⠀⣀⣬⣷⣾
  9. // ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡿⠟⣏⠀⠀⠀⠀⠀⠀⠀⠀⠈⢧⠀⠀⠀⠀⠘⡆⠀⠀⢀⣤⣾⣿⣿⣿⣿
  10. // ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⠿⠛⢻⡉⠀⠀⠈⢣⡀⠀⠀⠀⠀⠀⠀⠀⠘⡆⠀⠀⠀⠀⠘⣦⣶⣿⣿⣿⣿⣿⣿⣿
  11. // ⠉⠛⡿⠻⡿⠿⠿⠿⠿⠿⣿⠛⣿⠛⠉⠉⠁⠀⠀⠀⠀⢿⡦⣀⠀⠀⠑⢄⠀⠀⠀⣀⣀⠀⠀⢹⡄⠀⠀⠀⢾⣿⣿⣿⣿⣿⣿⣿⣿⣿
  12. // ⠀⠀⡇⠀⡇⠀⠀⠀⠀⠀⡏⠀⣸⢣⠀⠀⠀⠀⠀⠀⠀⠀⢳⣬⣷⣦⣤⡴⠷⢞⠋⠉⠀⠀⠀⠀⢧⠀⠀⠀⢸⣿⣿⣿⣿⣿⣿⣿⣿⣿
  13. // ⠀⠀⢰⠀⣷⠀⠀⠀⢀⣀⣇⣤⡿⠶⠳⡄⠀⠀⠀⠀⠀⠀⠈⠙⢄⠀⠈⠑⠢⢄⡙⠦⣄⠀⠀⠀⠸⡆⠀⠀⠀⣿⣿⣿⣿⣿⣿⣿⣿⣿
  14. // ⣀⣀⣘⡆⢹⣇⠀⠀⠀⠀⣯⢠⡇⠀⠀⠘⢦⡀⠀⠀⠀⠀⠀⢦⣈⠳⣄⠀⠀⠀⠀⠉⠛⠛⠲⣄⣀⣷⢸⠃⠀⣿⣿⣿⣿⣿⣿⣿⣿⣿
  15. // ⣿⣿⣿⣿⡈⢿⣆⠀⠀⠀⢹⣼⠀⠀⠀⠀⠀⠙⠳⣄⡀⠀⠀⠀⠙⢿⣲⣷⣤⣶⡶⠾⠿⠿⢿⡛⠋⠹⣾⡀⠀⢿⣿⣿⣿⣿⣿⣿⣿⡿
  16. // ⣿⣿⣿⣿⡇⠈⢯⠂⠀⢀⡬⣧⠴⡶⠶⠶⣤⠄⠀⠀⠙⠒⠦⠤⠤⣤⣍⣻⣥⡇⡆⠀⡴⣤⠀⢷⡀⠀⡏⡇⠀⠘⣿⣿⣿⣿⡿⠟⠋⠀
  17. // ⣿⣿⣿⣿⣷⠀⠈⢳⡀⠀⢠⢸⡀⢷⠀⣠⢬⠳⣄⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⢷⡁⣀⠷⠞⠀⣼⠇⠀⢀⡇⠀⠀⣼⣿⠋⠁⠀⠀⠀⠀
  18. // ⣿⣿⣿⣿⣿⡄⠀⠀⡟⢆⠈⠳⣷⡘⣧⡨⠟⠃⢺⠆⠀⠀⠀⠀⠀⠀⠀⠀⠀⠈⠛⠓⠒⠒⠛⠉⠀⢀⡞⢀⣠⣾⣿⣿⠀⠀⠀⠀⠀⠀
  19. // ⣿⣿⣿⣿⣿⡇⠀⢸⠀⠀⠙⢦⣙⢇⠈⠛⠲⠒⠋⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠐⣻⣞⠋⠀⢹⣿⡿⠀⠀⠀⠀⠀⣰
  20. // ⠿⠿⠿⠛⠙⡇⢠⡏⠀⠀⠀⠀⢿⣟⠳⣄⠀⠀⠀⠀⠀⠀⢀⡀⠀⠀⢠⡤⠦⠦⠶⢆⠀⠀⠀⣠⡞⢹⡇⠀⠀⠸⣿⡇⠀⠀⠀⠀⢠⠃
  21. // ⠀⠀⠀⠀⢸⣡⠋⠘⡄⠀⠀⠀⢸⡌⣳⣄⠀⠀⠀⠀⠀⠀⢘⠉⠉⠁⠀⠀⠀⠀⠀⠘⣇⣤⠞⠁⢠⠎⡇⠀⠀⠀⣿⡇⠀⠀⠀⠀⠀⠀
  22. // ⠀⠀⠀⠀⠚⠉⡆⠀⢹⡀⠀⠀⠀⣷⣿⣿⣗⢦⣤⣄⣀⣀⣸⣆⣤⣤⣤⣤⣤⠤⠶⠛⠛⠒⠦⡀⡎⠀⢸⠀⠀⠀⣿⣧⠀⠀⠀⠀⠀⠀
  23. // ⠀⠀⠀⠀⠀⠀⢧⠀⠀⢳⠀⠀⠀⢻⣿⣿⣿⣆⠀⠀⠸⣿⣿⣿⣿⣿⣿⠀⠹⣆⠀⠀⠀⣀⠀⢹⠇⣀⢸⡄⠀⠀⣿⣿⠀⠀⠀⠀⠀⠀
  24. // ⠀⠀⠀⠀⠀⠀⢸⠀⢠⠾⡄⠀⠀⢸⣿⣿⣿⣿⣧⠀⠀⠹⣿⣿⣿⣿⣿⠀⠀⠘⠳⠀⠘⠿⠿⢿⠿⠟⠸⡇⠀⠀⢿⣿⣧⠀⠀⠀⠀⠀
  25. // ⡀⡀⠀⠀⠀⠀⠈⣦⠋⠀⣧⠀⠀⢸⣿⣿⣿⣿⣿⣆⠀⠀⢻⣿⣿⣿⣿⣧⠀⠀⠀⠀⠀⠀⠀⣾⡄⠀⠀⡇⠀⠀⢸⣿⣿⣷⣄⣀⣠⣴
  26.  
  27. // ------- << AUTHOR: PHAM GIA KIET >> ------------
  28. // ようこそ、私のソウルソサエティへ
  29.  
  30. #include<iostream>
  31. #include<vector>
  32. #include<queue>
  33. #include<algorithm>
  34.  
  35. using namespace std;
  36.  
  37.  
  38. #define pb push_back
  39. #define  nhentai "virus"
  40. #define vlxx \
  41.     ios_base::sync_with_stdio(0); \
  42.     cin.tie(0); \
  43.     cout.tie(0)
  44. #define FOR(i, a, b) for (int i = (a); i <= (b); i++)
  45. #define ll long long
  46. #define ii pair<int, int>
  47. #define iii pair<ll, ii>
  48. #define fi first
  49. #define se second
  50.  
  51.  
  52. const int mn = 1e5+5;
  53. const ll INF = 1e18;
  54.  
  55. vector<int> adj1[mn], adj2[mn];
  56. int tin2[mn]; int tout2[mn];
  57. int times, n;
  58.  
  59. int bit[mn];
  60.  
  61. void upd(int idx, int val){
  62.     while(idx<=n){
  63.         bit[idx]+=val;
  64.         idx+=idx&-idx;
  65.     }
  66. }
  67.  
  68. int get(int idx){
  69.     int sum = 0;
  70.     while(idx>0){
  71.         sum+= bit[idx];
  72.         idx-=idx&-idx;
  73.     }
  74.     return sum;
  75. }
  76.  
  77. void dfs2(int u) {
  78.     times++;
  79.     tin2[u] = times;
  80.     for (int v : adj2[u]) {
  81.         dfs2(v);
  82.     }
  83.     tout2[u] = times;
  84. }
  85.  
  86. ll ans=0;
  87.  
  88. void dfs1(int u){
  89.     int precnt = get(tout2[u]) - get(tin2[u] - 1);
  90.  
  91.  
  92.     for (int v : adj1[u]){
  93.         dfs1(v);
  94.     }
  95.     int cntfinal = get(tout2[u]) - get(tin2[u] - 1);
  96.     ans += (cntfinal - precnt);
  97.     upd(tin2[u], 1);
  98. }
  99.  
  100. signed main(){
  101.     vlxx;
  102.     if(fopen(nhentai".inp", "r")){
  103.         freopen(nhentai".inp","r",stdin);
  104.         freopen(nhentai".out","w",stdout);
  105.     }
  106.     cin>>n;
  107.     int root1=0, root2=0;
  108.     FOR(i,1,n){
  109.         int p; cin>>p;
  110.         if (p==0) root1=i;
  111.         else adj1[p].pb(i);
  112.     }
  113.     FOR(i,1,n){
  114.         int p; cin>>p;
  115.         if (p==0) root2=i;
  116.         else adj2[p].pb(i);
  117.     }
  118.      times = 0;
  119.     dfs2(root2);
  120.     dfs1(root1);
  121.  
  122.     cout << ans << endl;
  123.  
  124.     return 0;
  125. }
Success #stdin #stdout 0.01s 8508KB
comments (?)
stdin 
Standard input is empty
stdout 
0
https://ideone.com/B6xwzM
language:
C++ (gcc 8.3)
created:
2 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!