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