1222. 可以攻击国王的皇后-力扣双百代码

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]]

解题代码如下：

/**
 * Return an array of arrays of size *returnSize.
 * The sizes of the arrays are returned as *returnColumnSizes array.
 * Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
 */
int** queensAttacktheKing(int** queens, int queensSize, int* queensColSize, int* king, int kingSize, int* returnSize, int** returnColumnSizes){
    int d[8];
    int r[8][2];
    int rz[8];
    int q;
    int rol=king[0],col=king[1];
    *returnColumnSizes=(int *)malloc(sizeof(int)*12);
    int i;
    for(i=0;i<8;i++){
        d[i]=10000;
        rz[i]=0;
    }
   // printf("--%d %d",rol,col);
    for(i=0;i<queensSize;i++){
        int x=queens[i][0],y=queens[i][1];
        q=0;
       // printf("%d %d ",x,y);
        if(x==rol&&y<col){
            if(abs(y-col)<d[q]){
                d[q]=abs(y-col);
                rz[q]=1;
                r[q][0]=x;
                r[q][1]=y;

            }
        }
        q++;
        if(x==rol&&y>col){
         //   printf("fa asfd%d dpp %d ",y,d[q]);
            if(abs(y-col)<d[q]){
                 d[q]=abs(y-col);
                rz[q]=1;
                r[q][0]=x;
                r[q][1]=y;

            }
        }
         q++;
        if(x>rol&&y==col){
        //    printf("-- %d %d %d",x,y,abs(x-rol));
         //   printf("| %d ",d[q]);
            if(abs(x-rol)<d[q]){
                 d[q]=abs(x-rol);
                rz[q]=1;
                r[q][0]=x;
                r[q][1]=y;

            }
        }
         q++;
        if(x<rol&&y==col){
            if(abs(x-rol)<d[q]){
                 d[q]=abs(x-rol);
                rz[q]=1;
                r[q][0]=x;
                r[q][1]=y;

            }
        }
         q++;
        if(x<rol&&y<col&&abs(x-rol)==abs(y-col)){
            if(abs(y-col)<d[q]){
                 d[q]=abs(y-col);
                rz[q]=1;
                r[q][0]=x;
                r[q][1]=y;

            }
        }
         q++;
        if(x>rol&&y<col&&abs(x-rol)==abs(y-col)){
            if(abs(y-col)<d[q]){
                 d[q]=abs(y-col);
                rz[q]=1;
                r[q][0]=x;
                r[q][1]=y;

            }
        }
        q++;
         if(x>rol&&y>col&&abs(x-rol)==abs(y-col)){
            if(abs(y-col)<d[q]){
                 d[q]=abs(y-col);
                rz[q]=1;
                r[q][0]=x;
                r[q][1]=y;

            }
        }
           q++;
        if(x<rol&&y>col&&abs(x-rol)==abs(y-col)){
            if(abs(y-col)<d[q]){
                 d[q]=abs(y-col);
                rz[q]=1;
                r[q][0]=x;
                r[q][1]=y;

            }
        }
       




    }
  
   int size=0;
        for(i=0;i<8;i++){
            if(rz[i]==1){
          queens[size][0]=r[i][0];
          queens[size][1]=r[i][1];
          (*returnColumnSizes)[size]=2;
        size++;
   
            }
        }

 
  // printf("%d ",size);
*returnSize=size;
 return queens;

}