华为OD 围棋的气

题目描述：

围棋棋盘由纵横各19条线垂直相交组成，棋盘上一共19x19=361个交点，对弈双方一方执白棋，一方执黑棋，落子时只能将棋子置于交点上。

“气”是围棋中很重要的一个概念，某个棋子有几口气，是指其上下左右方向四个相邻的交叉点中，有几个交叉点没有棋子，由此可知：

1、在棋盘的边缘上的棋子最多有3口气（黑1），在棋盘角点的棋子最多有2口气（黑2），其它情况最多有4口气（白1）

2、所有同色棋子的气之和叫作该色棋子的气，需要注意的是，同色棋子重合的气点，对于该颜色棋子来说，只能计算一次气，比如下图中，黑棋一共4口气，而不是5口气，因为黑1和黑2中间红色三角标出的气是两个黑棋共有的，对于黑棋整体来说只能算一个气。

3、本题目只计算气，对于眼也按气计算，如果您不清楚“眼”的概念，可忽略，按照前面描述的规则计算即可。

现在，请根据输入的黑棋和白棋的坐标位置，计算黑棋和白起一共各有多少气？

输入描述：

输入包括两行数据，如：

0 5 8 9 9 10

5 0 9 9 9 8

1、每行数据以空格分隔，数据个数是2的整数倍，每两个数是一组，代表棋子在棋盘上的坐标；

2、坐标的原点在棋盘左上角点，第一个值是行号，范围从0到18；第二个值是列号，范围从0到18。

3、举例说明：第一行数据表示三个坐标（0，5）、（8，9）、（9，10）

4、第一行表示黑棋的坐标，第二行表示白棋的坐标。

5、题目保证输入两行数据，无空行且每行按前文要求是偶数个，每个坐标不会超出棋盘范围。

输出描述：

8 7

两个数字以空格分隔，第一个数代表黑棋的气数，第二个数代表白棋的气数。

补充说明：

示例1

输入：

0 5 8 9 9 10
5 0 9 9 9 8

输出：

8 7

说明：

如果将输入数据放到棋盘上，数数黑棋一共8口气：

数数白棋一共7口气：

代码 

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String[] hei=sc.nextLine().split(" ");//黑棋的坐标
        String[] bai=sc.nextLine().split(" ");//白棋的坐标
        Set<WeiQi> set=new HashSet<>();//棋盘上存在的坐标
        WeiQi weiQi=new WeiQi();
        //黑棋加入棋盘
        for(int i=0;i<hei.length;i=i+2){
            int x=Integer.parseInt(hei[i]);
            int y=Integer.parseInt(hei[i+1]);
            weiQi=new WeiQi(x,y);
            boolean add = set.add(weiQi);
        }
        //白棋加入棋盘
        for(int i=0;i<bai.length;i=i+2){
            int x=Integer.parseInt(bai[i]);
            int y=Integer.parseInt(bai[i+1]);
            weiQi=new WeiQi(x,y);
            boolean add = set.add(weiQi);
        }
        //气=总数-棋盘上的棋子数
        int h=f(hei,set)- set.size();
        int b=f(bai,set)- set.size();
        System.out.println(h+" "+b);

    }
    //棋盘上的旗子数加上气，set排除相同的位置
    public static int f(String[] str,Set<WeiQi> set){
        Set<WeiQi> set2=new HashSet<>(set);
       for(int i=0;i<str.length;i=i+2){
           int x=Integer.parseInt(str[i]);
           int y=Integer.parseInt(str[i+1]);
           if(x-1>0){
               set2.add(new WeiQi(x-1,y));
           }
           if(x+1<19){
               set2.add(new WeiQi(x+1,y));
           }
           if(y-1>0){
               set2.add(new WeiQi(x,y-1));
           }
           if(y+1<19){
               set2.add(new WeiQi(x,y+1));
           }
       }
       return set2.size();
    }
}
class WeiQi{
    int x;
    int y;

    WeiQi(){

    }
    WeiQi(int x, int y){
        this.x=x;
        this.y=y;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        WeiQi weiQi = (WeiQi) o;
        return x == weiQi.x && y == weiQi.y;
    }

    @Override
    public int hashCode() {
        return Objects.hash(x, y);
    }


}

补充

1. 将所有的黑白棋子加入set（因为棋子在的位置不算气，一个围棋的上下左右可能存在其他围棋）；
2. 在将气加入set；
3. set包含所有的棋子以及气，将总数减去棋子数极为气。