每日一题~leetcode

1706. 球会落何处

题目大意

• 每个单元格都有一个对角线挡板，可以将球导向左侧或者右侧
• 将球导向右侧的挡板跨过左上角和右下角，在网格中用 1 表示
• 将球导向左侧的挡板跨过右上角和左下角，在网格中用 -1 表示
• 返回一个大小为 n 的数组 answer ，其中 answer[i] 是球放在顶部的第 i 列后从底部掉出来的那一列对应的下标，如果球卡在盒子里，则返回 -1

解题思路

1. 球向下滑动的位置（i , j)

将球导向右侧的挡板	右下方 (i+1 , j+1)
将球导向左侧的挡板	左下方 (i+1 , j+1)

2. 球是能下滑

可以向右下滑	grid (i, j) == grid (i , j + 1)
可以向左下滑	grid (i, j) == grid (i , j - 1)
解释	向右下滑必须满足当前位置和它右侧位置的挡板同向，同理，向左下滑必须满足当前位置和它左侧的挡板方向一致

3. 球不能下滑

球不能下滑	两个挡板或者挡板与樯形成 v 字形

代码

int* findBall(int** grid, int gridSize, int* gridColSize, int* returnSize){
    int *ans = (int*)malloc(sizeof(int)*gridColSize[0]);   //定义一个长度为n的数组
    int col = 0;
    for(int i = 0; i < gridColSize[0]; i++){     //遍历每一个可以放球的位置
        col = i;            // 球的初始位置
        for(int j = 0; j < gridSize; j++){       //模拟球滑行轨迹
            int dir = grid[j][col];        //当前板的方向      
            col += dir;             //col表示将要滑向的方向,球将要滑向的下一个位置{ 假设球可以向下滑，如果是向右的挡板，则下一个滑行位置一定是当前位置的右下方，同理，如果是向左的挡板，则下一个滑行位置一定是当前位置的左下方 }
            if(col < 0 || col == gridColSize[0] || grid[j][col] != dir){   //col < 0  到达左挡板，与左挡板形成v字形
                col = -1;                                                  //col == gridColSize[0] 到达右挡板，与右挡板形成v字形
                break;                                                     //grid[j][col] != dir 成 V 字形（将要滑向
            }                           //         的位置的上方的挡板不是和当前位置的挡板不是同一个方向，即形成 V 字形）
        }
        ans[i] = col;
    }
    *returnSize = gridColSize[0];
    return ans;
}