计蒜之道初赛第一场-阿里的新游戏

阿里的新游戏

阿里九游开放平台近日上架了一款新的益智类游戏——成三棋。成三棋是我国非常古老的一个双人棋类游戏，其棋盘如下图所示：

成三棋的棋盘上有很多条线段，只能在线段交叉点上放入棋子。我们可以用坐标系来描述棋盘：

如果一条线段上的三个交叉点都被同一玩家的棋子占据的话，则称这条线段被该玩家 成三。现在，小红和小明两人在游戏平台上下棋，其中小红的棋子是黑色的。请你帮小红计算他成三的线段数。

样例对应的棋盘如下：

输入格式

输入第一行两个整数 n,m(3 \le n, m \le 9)n,m(3≤n,m≤9)，nn 表示小红的棋子数，mm 表示小明的棋子数。

接下来 nn 行输入小红的棋子坐标。

接下来 mm 行输入小明的棋子坐标。

输入保证坐标合法，并且棋子之间不重合。

输出格式

输出小红成三的线段数。

样例输入

6 3
-1 0
-2 0
-3 0
-1 -1
-1 1
1 0
0 2
0 3
2 2
样例输出

2

题目大意：有一个棋盘（图见链接），给出棋子分布，问有多少个“成三”。
解题思路：各种姿势应该都能过，直接枚举或排序判断等等，代码写得丑。。。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<algorithm>
#include<map>
#include<string>
#include<cstring>
#include<vector>
#include<set>
#include<cmath>
using namespace std;

struct node
{
    int x,y;
}a[30],b[30];

bool cmp1(node u,node v)
{
    if(u.y==v.y) return u.x<v.x;
    return u.y<v.y;
}

bool cmp2(node u,node v)
{
    if(u.x==v.x) return u.y<v.y;
    return u.x<v.x;
}

int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        for(int i=1;i<=n;i++)
        {
            cin>>a[i].x>>a[i].y;
        }
        for(int i=1;i<=m;i++)
        {
            cin>>b[i].x>>b[i].y;
        }
        int ans=0;
        sort(a+1,a+n+1,cmp1);
        for(int i=1;i<=n;i++)
        {
            if(a[i].y==0&&i+2<=n)
            {
                if(a[i+2].y==0&&a[i+2].x-a[i].x==2)
                {
                    ans++;
                    i=i+2;
                }
            }else if(i+2<=n&&a[i].y==a[i+2].y)
            {
                ans++;
                i=i+2;
            }
        }
        sort(a+1,a+n+1,cmp2);
        for(int i=1;i<=n;i++)
        {
            if(a[i].x==0&&i+2<=n)
            {
                if(a[i+2].x==0&&a[i+2].y-a[i].y==2)
                {
                    ans++;
                    i=i+2;
                }
            }else if(i+2<=n&&a[i].x==a[i+2].x)
            {
                ans++;
                i=i+2;
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}