fork download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. #define read(type) readInt<type>() // Fast read
  4. #define ll long long
  5. #define nL "\n"
  6. #define pb push_back
  7. #define mk make_pair
  8. #define pii pair<int, int>
  9. #define a first
  10. #define b second
  11. #define vi vector<int>
  12. #define all(x) (x).begin(), (x).end()
  13. #define umap unordered_map
  14. #define uset unordered_set
  15. #define MOD 1000000007
  16. #define imax INT_MAX
  17. #define imin INT_MIN
  18. #define exp 1e9
  19. #define sz(x) (int((x).size()))
  20.  
  21. void solve() {
  22.     int boxNum, ops;
  23.     cin >> boxNum >> ops;
  24.  
  25.     int ballNum = ops;
  26.  
  27.     map<int, vector<int>> boxes;
  28.     for (auto i = 0; i < boxNum; i++) {
  29.         boxes[i].push_back(i);
  30.     }
  31.  
  32.     for (auto i = 0; i < ops; i++) {
  33.         int type, x, y;
  34.         cin >> type >> x;
  35.  
  36.         if (type == 1) {
  37.             cin >> y;
  38.             for (auto j = 0; j < boxes[y].size(); j++) {
  39.                 int val = boxes[y][j];
  40.                 boxes[y].erase(boxes[y].begin() + j);
  41.                 boxes[x].push_back(val);
  42.             }
  43.         } else if (type == 2) {
  44.             int ball = ballNum+1;
  45.             ballNum++;
  46.             boxes[x].push_back(ball);
  47.         } else {
  48.             for (auto const& dataPair : boxes) {
  49.                 for (auto j=0; j < dataPair.second.size(); j++) {
  50.                     if (dataPair.second[j] == x) {cout << x << endl;}
  51.                 }
  52.             }
  53.         }
  54.  
  55.  
  56.     }
  57. }
  58.  
  59.  
  60. int32_t main()
  61. {
  62.     ios_base::sync_with_stdio(false);
  63.     cin.tie(NULL);
  64.     solve();
  65.     return 0;
  66. }
Success #stdin #stdout 0.01s 5516KB
comments (?)
stdin
copy 
5 10
3 5
1 1 4
2 1
2 4
3 7
1 3 1
3 4
1 1 4
3 7
3 6
stdout
copy 
4
https://ideone.com/pGtEAh
language:
C++14 (gcc 8.3)
created:
1 year 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!