Educational Codeforces Round 119 (Rated for Div. 2) E. Replace the Numbers正难则反

https://codeforces.com/problemset/problem/1620/E

• 容易想到的是最后数列的长度就是操作1的次数，但是我想从前往后看，遇到了困难，我也想到了使用并查集，但没有反向考虑
• 因为正向的话由于后面的元素还没有出现，那么如果直接使用并查集那么会产生错误覆盖，所以反向考虑这个问题，因为后面的元素出现了那前面也一定出现了，所以每出现一次操作1，把这个数字对应的祖先节点加入到数组中；如果是操作2，那么就把$s[x]=s[y]$，$y$的祖先节点给$x$
• 如果正向有$1->3,3->5$，那么实际上就是前面的$1$最后都变成了$5$，但是后面的$1$还不能变，这样逆向考虑之后由于$3$的祖先节点已经变成了$5$，所以直接让$1$的祖先节点也变成$5$就好了
• 这个题的思路很经典

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

int main(){
  ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  int q;
  cin >> q;
  map<int, vector<int> > mp;
  int mx = -1;
  for(int i=1;i<=q;i++){
    int opt, x, y;
    cin >> opt;
    if(opt == 1){
      cin >> x;
      mp[i].push_back(x);
      mx = max(mx, x);
    }else{
      cin >> x >> y;
      mp[i].push_back(x);
      mp[i].push_back(y);
      mx = max({mx, x, y});
    }
  }
  vector<int> ans;
  vector<int> s(mx + 1);
  iota(s.begin(), s.end(), 0);
  for(int i=q;i>=1;i--){
    if(mp[i].size() == 1){
      int x = mp[i][0];
      ans.push_back(s[x]);
    }else{
      int x = mp[i][0];
      int y = mp[i][1];
      s[x] = s[y];
    }
  }
  reverse(ans.begin(), ans.end());
  for(int i=0;i<ans.size();i++){
    cout << ans[i] << " \n"[i == ans.size() - 1];
  }
  return 0;
}