STL之map、set灵活使用

1、LA 5908/UVA1517 Tracking RFIDs

　　题意：给出s个传感器的位置，以及其感应范围。如果某个方向上有墙，则该方向上感应距离减1.现在有w个墙，给出p个物品的位置，问其能被几个传感器探测到。

　　思路：传感器用set保存。之后传感器和物品之间是否有墙用线段是否相交来判断。

#include<iostream>
#include<algorithm>
#include<vector>
#include<set>
#include<cmath>
using namespace std;
typedef long long ll;
const double eps = 1e-6;

struct point
{
    int x, y;
    point(int xx=0, int yy=0):x(xx),y(yy){ }
    friend bool operator < (const point& a,const point& b)
    {
        return (a.x != b.x) ? a.x < b.x : a.y < b.y;
    }
};
struct line
{
    int x;
    int y;
    line(int xx=0,int yy=0):x(xx),y(yy){ }
};

line operator - (point a, point b)
{//得到点a和点b形成的直线的方向向量
    return line(a.x - b.x, a.y - b.y);
}
int cross(line a, line b)
{//返回0则两条直线平行，否则，直线相交:大于0说明b.y/b.x>a.y/a.x,b的斜率大
    return a.x * b.y - b.x * a.y;
}
double dis(point a, point b)
{//两点之间的直线距离
    return sqrt((a.x - b.x)*(a.x - b.x) + (a.y - b.y)*(a.y - b.y));
}

int direction(point p1, point p2, point p3)
{//判断点p3在直线p1-p2上还是在其两侧，上侧为负，下侧为正，在直线上为0
    return cross(p3 - p1, p2 - p1);
}

//判断点np是否在线段p1-p2
bool on_segment(point p1, point p2, point np)
{
    if (np.x < min(p1.x, p2.x)) return false;
    if (np.x > max(p1.x, p2.x)) return false;
    if (np.y < min(p1.y, p2.y)) return false;
    if (np.y > max(p1.y, p2.y)) return false;
    return true;
}

//判断线段p1-p2和p3-p4是否相交
bool segment_intersect(point p1, point p2, point p3, point p4)
{
    // 必须用 long long
    ll d1 = direction(p1, p2, p3);
    ll d2 = direction(p1, p2, p4);
    ll d3 = direction(p3, p4, p1);
    ll d4 = direction(p3, p4, p2);

    if (d1 * d2 < 0 && d3 * d4 < 0) return true;//p1、p2分别在线段p3-p4两侧，并且p3、p4分别在线段p1-p2两侧则说明两条线段相交
    if (d1 == 0 && on_segment(p1, p2, p3)) return true;
    if (d2 == 0 && on_segment(p1, p2, p4)) return true;
    if (d3 == 0 && on_segment(p3, p4, p1)) return true;
    if (d4 == 0 && on_segment(p3, p4, p2)) return true;
    return false;
}

set<int> sensors[20005];//传感器坐标
set<int>::iterator it;
point wall_l[15], wall_r[15];//墙的起止点坐标

int main()
{
    int T;
    scanf("%d", &T);

    while (T--)
    {
        for (int i = 0; i < 20005; i++)
            sensors[i].clear();

        int s, r, w, p;
        scanf("%d%d%d%d", &s, &r, &w, &p);

        for (int i = 0; i < s; i++)
        {
            int x, y;
            scanf("%d%d", &x, &y);
            sensors[x + 10000].insert(y);//注意x为负数处理
        }

        for (int i = 0; i < w; i++)
            scanf("%d%d%d%d", &wall_l[i].x, &wall_l[i].y, &wall_r[i].x, &wall_r[i].y);

        while (p--)
        {
            point p1;
            scanf("%d%d", &p1.x, &p1.y);

            vector<point> ans;
            for (int i = -r; i <= r; i++)
            {//枚举x差值
                int x = p1.x + i;
                int y = ceil(p1.y - sqrt(r*r - (p1.x - x)*(p1.x - x)));//确定当前x坐标y可取下限
                int k = x + 10000;
                if (k < 0 || k > 20000) continue;
                it = sensors[k].lower_bound(y);
                while (it != sensors[k].end())
                {//传感器的横坐标为x，从当前y值增加
                    point p2(x, *it++);//当前可取传感器
                    double d = dis(p1, p2);
                    if (d > r + eps) break;

                    double add = 0;
                    for (int k = 0; k < w; k++)//判断是否有墙在两点之间
                        if (segment_intersect(p1, p2, wall_l[k], wall_r[k])) add++;

                    if (d + add < r + eps) ans.push_back(p2);
                }
            }
            sort(ans.begin(), ans.end());
            printf("%d", ans.size());
            for (int i = 0; i < ans.size(); i++)
                printf(" (%d,%d)", ans[i].x, ans[i].y);
            printf("
");
        }
    }
    return 0;
}

 2、 LA 3634/UVA 12096 The SetStack Computer

　　题意：模拟集合栈。

　　思路：给每一个不同集合映射一个编号。

#include <iostream>
#include <set>
#include <map>
#include <vector>
#include <stack>
#include <algorithm>
#include<iterator>
using namespace std;

map<set<int>, int> id;
map<set<int>, int>::iterator it;
vector<set<int> > v;
char op[15];
int getID(set<int> x)
{
    it = id.find(x);
    if (it != id.end())
        return it->second;
    v.push_back(x);
    return id[x] = v.size() - 1;
}

int main()
{
    int T, n;
    scanf("%d", &T);
    while (T--)
    {
        scanf("%d", &n);
        id.clear();
        v.clear();
        stack<int> s;
        while (n--)
        {
            scanf("%s", op);
            if (op[0] == 'P')
                s.push(getID(set<int>()));
            else if (op[0] == 'D')
                s.push(s.top());
            else
            {
                set<int> t1 = v[s.top()]; s.pop();
                set<int> t2 = v[s.top()]; s.pop();
                set<int> t;
                if (op[0] == 'U')
                {
                    set_union(t1.begin(), t1.end(), t2.begin(), t2.end(), inserter(t, t.begin()));
                    s.push(getID(t));
                }
                else if (op[0] == 'I')
                {
                    set_intersection(t1.begin(), t1.end(), t2.begin(), t2.end(), inserter(t, t.begin()));
                    s.push(getID(t));
                }
                else if (op[0] == 'A')
                {
                    t2.insert(getID(t1));
                    s.push(getID(t2));
                }
            }
            printf("%d
",v[s.top()].size());
        }
        printf("***
");
    }
    return 0;
}