UVa 12096 - The SetStack Computer

最新推荐文章于 2023-10-14 16:31:24 发布

原创最新推荐文章于 2023-10-14 16:31:24 发布 · 270 阅读

0 ·

CC 4.0 BY-SA版权

本文通过解析《算法竞赛入门经典第2版》中的一段代码，深入讲解了集合操作在算法竞赛中的应用。从集合的创建、并集、交集到差集操作，代码演示了如何使用C++标准库进行高效处理。通过实际案例，读者可以学习到如何在比赛中快速实现集合算法，提高解题效率。

请看rujia大神的《算法竞赛入门经典第2版》
P115~117

#include <cstdio>
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <algorithm>

#define ALL(x) x.begin(),x.end()
#define INS(x) inserter(x, x.begin())
using namespace std;
typedef set<int> myset;
map<myset, int> idcache;
vector<myset> setcache;
stack<int> s;
int get_id(myset x)
{
  if (idcache.count(x)) return idcache[x];
  setcache.push_back(x);
  return idcache[x] = setcache.size() - 1;
}
void solve()
{
  int n;
  cin >> n;
  for (int i = 0; i < n; i++)
  {
    string op;
    cin >> op;
    if (op[0] == 'P') s.push(get_id(myset())); else
    if (op[0] == 'D') s.push(s.top()); else
    {
      myset x1 = setcache[s.top()];
      s.pop();
      myset x2 = setcache[s.top()];
      s.pop();
      myset x;
      if (op[0] == 'U') set_union(ALL(x1), ALL(x2), INS(x));
      if (op[0] == 'I') set_intersection(ALL(x1), ALL(x2), INS(x));
      if (op[0] == 'A')
      {
        x = x2;
        x.insert(get_id(x1));
      }
      s.push(get_id(x));
    }
    cout << setcache[s.top()].size() << endl;
  }
  puts("***");
}
int main()
{
//  freopen("input.txt", "r", stdin);
  int t;
  cin >> t;
  for (int i = 0; i < t; i++)
    solve();
}