Codeforces Gym 100341I Hungry Queen 2 Set+模拟

题目大意：

在一个无穷大的棋盘上有一个皇后在(0,0)点，皇后可以上下左右或者沿对角线移动。有n个卒，皇后要按卒的输入顺序吃卒，每次移动必须吃掉一个卒，皇后不能跳过卒。问皇后最多能吃掉几个卒？

做法：

• 先将所有卒的位置(x、y、x+y、x-y)离散化；
• 用set分别记录，在同一行（按y排序）、同一列（按x排序）、同一对角线的卒；
• 模拟皇后的移动，每次在当前位置与下一个卒的位置所确定的set中查找下一个卒的前驱和后继，如果前驱和后继中有一个是当前位置，这说明在当前位置与下一个卒的方向上没有障碍，答案+1；
• 在所有方向的set中删掉当前点，然后移动到下一个点；
• 反之，如果下一个卒与当前位置不能构成合法移动，或者在当前位置与下一个卒的方向上有障碍，那么就结束模拟输出答案。

代码：

4551665 Accepted 24532 498 GNU G++ 4.9.2 2606 5:06:16

#include <bits/stdc++.h>using namespace std;const int maxn=100001;int x[maxn],y[maxn],lx[maxn],ly[maxn],lxpy[maxn],lxmy[maxn];struct cmp2{    bool operator ()(int a,int b)const    {        return x[a]<x[b];    }};//按x排序 struct cmp1{    bool operator ()(int a,int b)const    {        return y[a]<y[b];    }};//按y排序 set<int,cmp1>mx[maxn],mxpy[maxn],mxmy[maxn];set<int,cmp1>::iterator ix,ipx;set<int,cmp2>my[maxn];set<int,cmp2>::iterator iy,ipy;int rp(int num,int l,int a[]){    return lower_bound(a,a+l,num)-a;}//返回离散化下标 int main(){    freopen("queen2.in","r",stdin);#ifdef ONLINE_JUDGE    freopen("queen2.out","w",stdout);#endif    int n;    scanf("%d",&n);    for(int i=0;i<n;i++)    {        scanf("%d%d",&x[i],&y[i]);        lx[i]=x[i];        ly[i]=y[i];        lxpy[i]=x[i]+y[i];        lxmy[i]=x[i]-y[i];    }    sort(lx,lx+n);    sort(ly,ly+n);    sort(lxpy,lxpy+n);    sort(lxmy,lxmy+n);    int sx=unique(lx,lx+n)-lx;    int sy=unique(ly,ly+n)-ly;    int sxpy=unique(lxpy,lxpy+n)-lxpy;    int sxmy=unique(lxmy,lxmy+n)-lxmy;    //离散化     int px=0,py=0;    for(int i=0;i<n;i++){        mx[rp(x[i],sx,lx)].insert(i);        my[rp(y[i],sy,ly)].insert(i);        mxpy[rp(x[i]+y[i],sxpy,lxpy)].insert(i);        mxmy[rp(x[i]-y[i],sxmy,lxmy)].insert(i);    }    //将同类点加到set中     //模拟移动     if(px==x[0] || py == y[0] || px+py ==x[0]+y[0] || px-py == x[0]-y[0]){        int ans=1,ans2=0;        px=x[0];py=y[0];        for(int i=1;i<n;i++)        {            ans2=ans;               if(px == x[i])            {                int pos=rp(x[i],sx,lx);                ix=++mx[pos].find(i);                if(ix!=mx[pos].end() && (*ix) == (i-1))++ans;                ipx=mx[pos].find(i);                if( ipx!=mx[pos].begin() )                    if(*(--ipx) == (i-1))++ans;            }else if(py == y[i])            {                int pos=rp(y[i],sy,ly);                iy=++my[pos].find(i);                if(iy!=my[pos].end() && (*iy) == (i-1))++ans;                ipy=my[pos].find(i);                if( ipy!=my[pos].begin() )                    if(*(--ipy) == (i-1))++ans;            }else if(px+py == x[i]+y[i])            {                int pos=rp(x[i]+y[i],sxpy,lxpy);                ix=++mxpy[pos].find(i);                if(ix!=mxpy[pos].end() && (*ix) == (i-1))++ans;                ipx=mxpy[pos].find(i);                if( ipx!=mxpy[pos].begin() )                    if(*(--ipx) == (i-1))++ans;            }else if(px-py == x[i]-y[i])            {                int pos=rp(x[i]-y[i],sxmy,lxmy);                ix=++mxmy[pos].find(i);                if(ix!=mxmy[pos].end() && (*ix) == (i-1))++ans;                ipx=mxmy[pos].find(i);                if( ipx!=mxmy[pos].begin() )                    if(*(--ipx) == (i-1))++ans;            }            else break;            if(ans==ans2)break;            mx[rp(px,sx,lx)].erase(i-1);            my[rp(py,sy,ly)].erase(i-1);            mxpy[rp(px+py,sxpy,lxpy)].erase(i-1);            mxmy[rp(px-py,sxmy,lxmy)].erase(i-1);            px=x[i];py=y[i];        }        printf("%d\n",ans);    }else printf("0\n");    return 0;}