其他分享
首页 > 其他分享> > STL模拟题

STL模拟题

作者:互联网

A Where is the Marble?

Des
给出一个数组,排序后,进行查找,第一个大于等于该询问元素的位置。
Solution
使用二分函数lower_bound即可,或者自己手写二分,前提是排序。
Code
这里就不给代码了,过于easy。

B The SetStack Computer

Des
你有一个集合栈,支持以下操作:
PUSH:空集'{}'入栈
DUP: 把当前栈顶元素赋值一份入栈
UNION:出栈两个集合,把二者的并集入栈
INTERSECT:出栈两个集合,把二者的交集入栈
ADD:出栈两个集合,把先出栈的集合加到后出栈的集合中,把结果入栈
Solution
我们使用一个vector存放所有的集合,根据id存取,栈中就放集合编号,重复的集合放入栈,都是同一个编号,有效节约空间。

set_union()可以求两个集合的并集
set_intersection()可以求两个集合的交集

函数用法传送门
具体细节看代码
Code

#include <bits/stdc++.h>
using namespace std;
typedef set<int> st ;
map<st, int> mp;//记录集合的标号
vector<st> v;//存放集合
int get_id(const st& x)
{
    if(mp.count(x))
    {
        //cout << "getid work\n";
        return mp[x];
    }
    v.push_back(x);
    return mp[x] = v.size() - 1;
}
signed main()
{
    ios::sync_with_stdio(false), cin.tie(0);
    int T, q;
    cin >> T;
    st null, x;//空集合
    while(T-- && cin >> q)
    {
        stack<int> s;//栈中放集合的id,对应集合
        for(int i = 0; i < q; ++i)
        {
            string op;
            cin >> op;
            if(op[0] == 'P') s.push(get_id(null));
            else if(op[0] == 'D') s.push(s.top());
            else
            {
                const st& x1 = v[s.top()]; s.pop();
                const st& x2 = v[s.top()]; s.pop();
                x.clear();
                if(op[0] == 'U')
                    //set_union求两个集合的并集
                    set_union(x1.begin(), x1.end(), x2.begin(), x2.end(),
                              inserter(x, x.begin()));
                if(op[0] == 'I')
                    //set_intersection求两个集合的交集
                    set_intersection(x1.begin(), x1.end(), x2.begin(), x2.end(),
                                     inserter(x, x.begin()));
                if(op[0] == 'A') x = x2, x.insert(get_id(x1));
                s.push(get_id(x));
            }
            cout << v[s.top()].size() << "\n";
        }
        cout << "***\n";
    }
    return 0;
}

标签:set,STL,模拟题,x2,集合,x1,id,op
来源: https://www.cnblogs.com/-ytz/p/16578618.html