LeetCode 1222. 可以攻击国王的皇后（set）

输入：queens = [[0,1],[1,0],[4,0],[0,4],[3,3],[2,4]], 
king = [0,0]
输出：[[0,1],[1,0],[3,3]]
解释： 
[0,1] 的皇后可以攻击到国王，因为他们在同一行上。 
[1,0] 的皇后可以攻击到国王，因为他们在同一列上。 
[3,3] 的皇后可以攻击到国王，因为他们在同一条对角线上。 
[0,4] 的皇后无法攻击到国王，因为她被位于 [0,1] 的皇后挡住了。 
[4,0] 的皇后无法攻击到国王，因为她被位于 [1,0] 的皇后挡住了。 
[2,4] 的皇后无法攻击到国王，因为她和国王不在同一行/列/对角线上。

输入：queens = [[0,0],[1,1],[2,2],[3,4],[3,5],[4,4],[4,5]], 
king = [3,3]
输出：[[2,2],[3,4],[4,4]]

输入：queens = [[5,6],[7,7],[2,1],[0,7],[1,6],[5,1],[3,7],[0,3],[4,0],[1,2],[6,3],[5,0],[0,4],[2,2],[1,1],[6,4],[5,4],[0,0],[2,6],[4,5],[5,2],[1,4],[7,5],[2,3],[0,5],[4,2],[1,0],[2,7],[0,1],[4,6],[6,1],[0,6],[4,3],[1,7]], king = [3,4]
输出：[[2,3],[1,4],[1,6],[3,7],[4,3],[5,4],[4,5]]

提示：
1 <= queens.length <= 63
queens[0].length == 2
0 <= queens[i][j] < 8
king.length == 2
0 <= king[0], king[1] < 8
一个棋盘格上最多只能放置一枚棋子。

class Solution {
public:
    vector<vector<int>> queensAttacktheKing(vector<vector<int>>& queens, vector<int>& king) {
        vector<vector<int>> dir = {{1,0},{0,1},{0,-1},{-1,0},{1,1},{1,-1},{-1,1},{-1,-1}};
        int i,j,k;
        set<pair<int,int>> s;
        vector<vector<int>> pos;
        for(i = 0; i < queens.size(); i++)
        	s.insert(make_pair(queens[i][0], queens[i][1]));//查找集合
        for(k = 0; k < dir.size(); ++k)//8个方向
        {
        	i = king[0];
        	j = king[1];//起点
        	while(1)
        	{
        		i += dir[k][0];
        		j += dir[k][1];//走起
        		if(!(i>=0 && i<8 && j>=0 && j<8))
        			break;//出界
        		else
        		{
        			if(s.count(make_pair(i,j)))
        			{	//在界内，找到皇后
        				pos.push_back({i,j});
        				break;
        			}
        		}
        	}
        }
        return pos;
    }
};