fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. int n, a, tar, cnt;
  4. int main(){
  5. cin >> n;
  6. vector<vector<int>> adj(n, vector<int>(0));
  7. for(int i = 0; i < n; i++){
  8. cin >> a;
  9. if(a != -1) adj[a].push_back(i);
  10. }
  11. cin >> tar;
  12.  
  13. for(int i = 0; i < n; i++){
  14. if(i == tar) continue;
  15. for(int j = 0; j < adj[i].size(); j++){
  16. if(!adj[adj[i][j]].size() && adj[i][j] != tar) cnt++;
  17. }
  18. }
  19. cout << cnt << '\n';
  20. }
Success #stdin #stdout 0s 5316KB
stdin
9
-1 0 0 2 2 4 4 6 6
4
stdout
4