1222 可以攻击国王的皇后

题目描述:
在一个 8x8 的棋盘上,放置着若干「黑皇后」和一个「白国王」。
「黑皇后」在棋盘上的位置分布用整数坐标数组 queens 表示,「白国王」的坐标用数组 king 表示。
「黑皇后」的行棋规定是:横、直、斜都可以走,步数不受限制,但是,不能越子行棋。
请你返回可以直接攻击到「白国王」的所有「黑皇后」的坐标(任意顺序)。

示例 1:
在这里插入图片描述
输入: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] 的皇后无法攻击到国王,因为她和国王不在同一行/列/对角线上。

示例 2:
在这里插入图片描述
输入:queens = [[0,0],[1,1],[2,2],[3,4],[3,5],[4,4],[4,5]], king = [3,3]
输出:[[2,2],[3,4],[4,4]]

示例 3:
在这里插入图片描述
输入: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[i].length == 2
0 <= queens[i][j] < 8
king.length == 2
0 <= king[0], king[1] < 8
一个棋盘格上最多只能放置一枚棋子。

方法1:
主要思路:解题链接汇总
(1)使用unordered_set标识出皇后的位置,从king出发,向八个方向找皇后,找到一个,则跳出当前方向;

class Solution {
public:
    vector<vector<int>> queensAttacktheKing(vector<vector<int>>& queens, vector<int>& king) {
        unordered_set<int> st;
        for(vector<int>&q:queens){//标识皇后的位置
            st.insert(10*q[0]+q[1]);
        }
        int row=king[0],col=king[1];//从国王的位置开始出发
        vector<vector<int>> res;
        //向八个方向找
        for(int i=row+1;i<8;++i){
            if(st.count(i*10+col)){
                res.push_back({i,col});
                break;
            }
        }
        for(int i=row-1;i>=0;--i){
            if(st.count(i*10+col)){
                res.push_back({i,col});
                break;
            }
        }
        for(int i=col+1;i<8;++i){
            if(st.count(row*10+i)){
                res.push_back({row,i});
                break;
            }
        }
        for(int i=col-1;i>=0;--i){
            if(st.count(row*10+i)){
                res.push_back({row,i});
                break;
            }
        }
        for(int i=1;row+i<8&&col+i<8;++i){
            if(st.count((row+i)*10+col+i)){
                res.push_back({row+i,col+i});
                break;
            }
        }
        for(int i=1;row+i<8&&col-i>=0;++i){
            if(st.count((row+i)*10+col-i)){
                res.push_back({row+i,col-i});
                break;
            }
        }
        for(int i=1;row-i>=0&&col+i<8;++i){
            if(st.count((row-i)*10+col+i)){
                res.push_back({row-i,col+i});
                break;
            }
        }
        for(int i=1;row-i>=0&&col-i>=0;++i){
            if(st.count((row-i)*10+col-i)){
                res.push_back({row-i,col-i});
                break;
            }
        }
        return res;
    }
};

方法2:
主要思路:
(1)将代码整合重写;

class Solution {
public:
    vector<vector<int>> queensAttacktheKing(vector<vector<int>>& queens, vector<int>& king) {
        vector<vector<bool>> signs(8,vector<bool>(8,false));
        for(vector<int>&q:queens){
            signs[q[0]][q[1]]=true;
        }
        vector<pair<int,int>>dirs={{0,1},{0,-1},{1,0},{-1,0},{1,1},{-1,-1},{-1,1},{1,-1}};
        vector<vector<int>> res;
        for(int i=0;i<dirs.size();++i){
            for(int row=king[0],col=king[1];row>=0&&row<8&&col>=0&&col<8;row+=dirs[i].first,col+=dirs[i].second){
                if(signs[row][col]){
                    res.push_back({row,col});
                    break;
                }
            }
        }
        return res;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值