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-;

 struct point
 {
     int x, y;
     point(int xx=, int yy=):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=,int yy=):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 <  && d3 * d4 < ) return true;//p1、p2分别在线段p3-p4两侧，并且p3、p4分别在线段p1-p2两侧则说明两条线段相交
     if (d1 ==  && on_segment(p1, p2, p3)) return true;
     if (d2 ==  && on_segment(p1, p2, p4)) return true;
     if (d3 ==  && on_segment(p3, p4, p1)) return true;
     if (d4 ==  && on_segment(p3, p4, p2)) return true;
     return false;
 }

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

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

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

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

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

         for (int i = ; 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 + ;
                 if (k <  || k > ) 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 = ;
                     for (int k = ; 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 = ; i < ans.size(); i++)
                 printf(" (%d,%d)", ans[i].x, ans[i].y);
             printf("\n");
         }
     }
     return ;
 }

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[];
 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() - ;
 }

 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[] == 'P')
                 s.push(getID(set<int>()));
             else if (op[] == '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[] == 'U')
                 {
                     set_union(t1.begin(), t1.end(), t2.begin(), t2.end(), inserter(t, t.begin()));
                     s.push(getID(t));
                 }
                 else if (op[] == 'I')
                 {
                     set_intersection(t1.begin(), t1.end(), t2.begin(), t2.end(), inserter(t, t.begin()));
                     s.push(getID(t));
                 }
                 else if (op[] == 'A')
                 {
                     t2.insert(getID(t1));
                     s.push(getID(t2));
                 }
             }
             printf("%d\n",v[s.top()].size());
         }
         printf("***\n");
     }
     return ;
 }