fork download
  1. /* package whatever; // don't place package name! */
  2.  
  3. import java.util.*;
  4. import java.lang.*;
  5. import java.io.*;
  6.  
  7. /* Name of the class has to be "Main" only if the class is public. */
  8. class Ideone
  9. {
  10. public static void main (String[] args) throws java.lang.Exception
  11. {
  12. // your code goes here
  13. Scanner sc=new Scanner(System.in);
  14. int n=sc.nextInt();
  15. int a[]=new int[n+1];
  16. for(int i=1;i<=n;i++)
  17. a[i]=sc.nextInt();
  18. Map <Integer,Integer> m=new HashMap<>();
  19. int count=0;
  20. for(int i=1;i<=n;i++)
  21. {
  22. int rhs=a[a[a[i]]];
  23. if(m.containsKey(rhs))
  24. count+=m.get(rhs);
  25. m.put(rhs,m.getOrDefault(rhs,0)+1);
  26. }
  27. System.out.println(count);
  28. }
  29. }
Success #stdin #stdout 0.11s 54568KB
stdin
4
4 2 1 2
stdout
6