腐爛的橘子【leetcode每日一題】

在這裏插入圖片描述
這道題算是一個搜索的板子題了。多源的寬度優先搜索。
大致思路如下:
1.首先遍歷統計初始數組中有多少好橘子和壞橘子
如果沒有壞橘子,返回-1;
如果沒有好橘子,返回0;
2.當好橘子不爲0得時候,開始每輪的循環。時間數加1,然後裏面所有壞橘子感染周圍的好橘子。結束後判斷這一輪感染完好橘子數有沒有減,沒有返回1.有的話且不爲0繼續下一輪感染,最後就得到了感染時間
代碼如下:

int orangesRotting(int** grid, int gridRowSize, int *gridColSizes) {
    int good = 0, bad = 0, t = 0;
    for(int i = 0; i < gridRowSize; i++)
        for(int j = 0; j < gridColSizes[0]; j++)
        {
            if     (grid[i][j] == 1) good++;
            else if(grid[i][j] == 2) bad++; 
        }
    if(!good) return 0;
    if(!bad) return -1;
    int new_good = good;
    while(new_good) 
    {
        t++;
        for(int i = 0; i < gridRowSize; i++)
            for(int j = 0; j < gridColSizes[0]; j++)
                if(grid[i][j] == 2)
                {
                    if(i && grid[i - 1][j] == 1)    {grid[i-1][j]= 3; new_good--;}
                    if(j && grid[i][j - 1] == 1)    {grid[i][j-1] = 3; new_good--;}
                    if(i != gridRowSize - 1 && grid[i + 1][j] == 1)    
                        {grid[i+1][j]= 3; new_good--;}
                    if(j != gridColSizes[0] - 1 && grid[i][j + 1] == 1)
                        {grid[i][j+1] = 3; new_good--;}
                }
        for(int i = 0; i < gridRowSize; i++)
            for(int j = 0; j < gridColSizes[0]; j++)
                if(grid[i][j] == 3)  grid[i][j] = 2;
        if(new_good == good) return -1;//沒有新的橘子變壞
        good = new_good;
    }
    return t;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章