字节跳动0414笔试编程题

第一题

和leetcode994. 腐烂的橘子一样的，AC解答：

# -*- coding:utf-8 -*-
__author__ = 'ShawDa'

def get_ret(res):
    x, y, ret = len(res), len(res[0]), 0
    locs, stack = [[-1, 0], [0, -1], [0, 1], [1, 0]], []
    for i in range(x):
        for j in range(y):
            if res[i][j] == 2:
                stack.append((i, j, 0))
    while stack:
        i, j, ret = stack.pop(0)
        for loc in locs