leetcode周赛158第二题

这道题主要考察了边界条件的考虑和基本的vector操作

可以攻击国王的皇后

在一个 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[0].length == 2
  • 0 <= queens[i][j] < 8
  • king.length == 2
  • 0 <= king[0], king[1] < 8
  • 一个棋盘格上最多只能放置一枚棋子。

 我的思路:

维持一个大小为8的flag数组,用来记录8个方向上的皇后,遍历皇后数组,如果在当前方向上找到比flag中的皇后距离更近的皇后,就更新flag的值,最后返回flag,如果某一方向上的flag值没有改变说明该方向无解,最终统计时将其删掉。

//思路是:维持一个大小为8的flag数组,用来记录8个方向上的皇后,遍历皇后数组,如果在当前方向上找到比flag中的皇后距离更近的皇后,就更新flag的值,最后返回flag,如果某一方向上的flag值一直没有改变说明该方向无解。
//按方向遍历更好还是按皇后遍历更好?
class Solution {
public:
    vector<vector<int>> queensAttacktheKing(vector<vector<int>>& queens, vector<int>& king) {
        vector<vector<int>> flag;
        vector<int> m;
        m.push_back(16);
        m.push_back(16);
        for(int i=0;i<8;i++){
            flag.push_back(m);
        }
        for(int i=0;i<queens.size();i++){
            int j=judge(queens[i][0],queens[i][1],king[0],king[1]);
            if(j==8) continue;
            if(abs(flag[j][0]-king[0])>abs(queens[i][0]-king[0])||abs(flag[j][1]-king[1])>abs(queens[i][1]-king[1])){
                flag.erase(flag.begin()+j);
                flag.insert(flag.begin()+j,queens[i]);
            }
        }
        vector<vector<int>> res;
        for(auto i=flag.begin();i<flag.end();i++){
            if((*i)[0]==16) continue;
            res.push_back(*i);
        }
        return res;
    }
private:
    int judge(int q1,int q2,int k1,int k2){
        if(abs(q1-k1)==abs(q2-k2)||q1==k1||q2==k2){
            if(q2==k2&&q1<k1) return 0;
            else if(q2==k2&&q1>k1) return 4;
            else if(q1==k1&&q2>k2) return 2;
            else if(q1==k1&&q2<k2) return 6;
            else if(q1-k1<0&&q2-k2>0) return 1;
            else if(q1-k1>0&&q2-k2>0) return 3;
            else if(q1-k1>0&&q2-k2<0) return 5;
            else if(q1-k1<0&&q2-k2<0) return 7;
        }
        else{
            return 8;
        }
        return -1;
    }
};

讨论区的做法:

 按照方向遍历,每个方向的格子按照从近到远的顺序检索,每次遍历整个queens,如果该方向上找到返回的值,就停止检索,继续检索下一个方向,直到8个方向都检索完毕。

代码

class Solution {
    template <class T>//计算数组长度的函数
    int getArrayLen(T& array)
    {
        return (sizeof(array) / sizeof(array[0]));
    }
public:
    vector<vector<int>> queensAttacktheKing(vector<vector<int>>& queens, vector<int>& king) {
        int dx[]={-1,-1,0,1,1,1,0,-1};
        int dy[]={0,1,1,1,0,-1,-1,-1};//设置好x,y方向上每次要走的数组
        vector<vector<int>> res;
        for(int d=0;d<getArrayLen(dx);d++){//逐个方向遍历
            int x=king[0];
            int y=king[1];
            while(true){//没找到符合的queen就一直遍历下去直到找到或越界break
                x+=dx[d];
                y+=dy[d];
                if(x<0||x>7||y<0||y>7) break;
                bool flag=false;
                int i;
                for(i=0;i<queens.size();i++){
                    if(x==queens[i][0]&&y==queens[i][1]){
                        flag=true;
                        break;
                    }
                }
                if(flag==true){
                    res.push_back(queens[i]);
                    break;
                }
            }
        }
        return res;
    }
};

在上图这种情况下,所有八个方向都没有皇后,而无关的格子上都有皇后,第一种做法的复杂度是queens的长度=36,第二种做法的复杂度是:(3*5+4*3)*36=972,显然第一种做法更好。

当所有八个方向都有皇后时,并且合格的皇后均在后面出现时,第一种做法会不断将错误的皇后放进去,然后换出来,浪费了时间。

在上图中的情况下,第二种做法的时间复杂度是20+21+22+23+24+25+26+27=470,第一种做法:如果把每次换出换入的过程看成两次,那么总的复杂度是:(1+3*2)*3+(1+2*2)*5=46。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值