【每周一坑】矩阵旋转

def ratate(matrix):
    '''
    : type matrix : List(List(Int))
    : rtype: List(List(Int))
    '''
    # your code here
    
m1 = [[]]
m2 = [[1]]
m3 = [[i for i in range(3)] for j in range(3)]
m4 = [[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]]

assert ratate(m1) == [[]]
assert ratate(m2) == [[1]]
assert ratate(m3) == [[0, 0, 0], [1, 1, 1], [2, 2, 2]]
assert ratate(m4) == [[13, 9, 5, 1], [14, 10, 6, 2], [15, 11, 7, 3], [16, 12, 8, 4]]

def judge_arms(i):
    # 将该数转换为字符串　123 -> '123'
    tem = str(i)
    # 按照　n位正整数等于其各位数字的n次方　进行求和
    # sum = 1**3 + 2**3 + 3**3
    tem_sum = sum([int(j)**len(tem) for j in tem])
    # 判断求和之后的数是否和原数相等
    if i == tem_sum:
        return True

'''
寻找小于 1000 的阿姆斯特朗数
逐一遍历小于 1000 的数
判断是否为阿姆斯特朗数
'''
def arms(N):
    for i in range(1, N+1):
        if judge_arms(i):
            print(i)

arms(1000)

'''
附加题: 输入一个正整数，输出距离它最近的阿姆斯特朗数。
以指定的数为基准，同时向前向后寻找阿姆斯特朗数
找到即返回该数
'''
def near_arms(N):
    forward, backward = N, N
    while True:
        if judge_arms(forward):
            return forward
        elif judge_arms(backward):
            return backward
        forward += 1
        backward -= 1

r = near_arms(100000)
print(r)