LeetCode解法汇总1222. 可以攻击国王的皇后

目录链接:

力扣编程题-解法汇总_分享+记录-CSDN博客

GitHub同步刷题项目:

https://github.com/September26/java-algorithms

原题链接:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台


描述:

在一个 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
  • 一个棋盘格上最多只能放置一枚棋子。

解题思路:

* 解题思路:

* 从king出发,使用searchForward方法分别向8个方向分别尝试。如果有满足条件的,则返回。如果没有满足的,则返回空数组。

代码:

class Solution1222
{
public:
    vector<vector<int>> fowards = {
   
   {1, 0}, {1, 1}, {1, -1}, {0, 1}, {0, -1}, {-1, 0}, {-1, 1}, {-1, -1}};

    vector<int> searchForward(set<string> &queenSet, vector<int> king, vector<int> &forward)
    {
        while (king[0] >= 0 && king[1] >= 0 && king[0] < 8 && king[1] < 8)
        {
            king[0] = king[0] + forward[0];
            king[1] = king[1] + forward[1];
            if (queenSet.find(to_string(king[0]) + "_" + to_string(king[1])) != queenSet.end())
            {
                return vector<int>{king[0], king[1]};
            }
        }
        return vector<int>{};
    }

    vector<vector<int>> queensAttacktheKing(vector<vector<int>> &queens, vector<int> &king)
    {
        set<string> queenSet;
        for (int i = 0; i < queens.size(); i++)
        {
            queenSet.insert(to_string(queens[i][0]) + "_" + to_string(queens[i][1]));
        }
        vector<vector<int>> list;
        for (int i = 0; i < fowards.size(); i++)
        {
            vector<int> result = searchForward(queenSet, king, fowards[i]);
            if (result.size() > 0)
            {
                list.push_back(result);
            }
        }
        return list;
    }
};

猜你喜欢

转载自blog.csdn.net/AA5279AA/article/details/132887617