fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. long long int n;
  6. cin>>n;
  7. vector<long long int>cow;//ids
  8. vector<long long int>shuffle;//pattern
  9. vector<long long int>invshuffle;
  10. for(long long int i = 0; i<n; i++){
  11. long long int a;
  12. cin>>a;
  13. shuffle.push_back(a);
  14. }
  15. for(long long int i = 0; i<n; i++){
  16. long long int a;
  17. cin>>a;
  18. cow.push_back(a);
  19. }
  20. for(long long int i = 0; i<n; i++){
  21. auto a=find(shuffle.begin(), shuffle.end(),i+1);
  22. long long ind=distance(shuffle.begin(),a);
  23. invshuffle.push_back(ind);
  24. cout<<ind<<" ";
  25. }
  26. vector<long long int>v=cow;
  27. for(long long int i = 0; i<3; i++){
  28. v[i]=cow[invshuffle[i]];
  29. }
  30. for(long long int i = 0; i<n; i++){
  31. cout<<v[i]<<endl;
  32. }
  33.  
  34.  
  35. return 0;
  36. }
Success #stdin #stdout 0.01s 5284KB
stdin
5
1 3 4 5 2
1234567 2222222 3333333 4444444 5555555
stdout
0 4 1 2 3 1234567
5555555
2222222
4444444
5555555