checkio-pawn brotherhood

来源:互联网 发布:知乎成都审核招聘 编辑:程序博客网 时间:2024/06/05 17:11

这里写图片描述
题目意思是给定一个8*8的棋盘,同时给定8个棋子的位置。如果某一个棋子的位置可以由其他棋子一步到达,则认为该棋子安全。判断8个棋子中安全的棋子的个数。
如图所示,左边的安全棋子个数为6,右边的位1。

思路:遍历给定的集合(set)里面的所有元素,然后转换为list。然后利用ord 和chr 转换求出该棋子左下和右下的棋子的位置坐标。如果该坐标在原来的集合中,则num+1。最后返回num。
代码如下:

def safe_pawns(pawns):    num=0    for i in pawns:        a=list(i)        #b=a[0]+chr((ord(a[1])-1))        c=chr(ord(a[0])-1)+chr((ord(a[1])-1))        d=chr(ord(a[0])+1)+chr((ord(a[1])-1))        if c in pawns or d in pawns:            num+=1    return numif __name__ == '__main__':    #These "asserts" using only for self-checking and not necessary for auto-testing    assert safe_pawns({"b4", "d4", "f4", "c3", "e3", "g5", "d2"}) == 6    assert safe_pawns({"b4", "c4", "d4", "e4", "f4", "g4", "e5"}) == 1

贴一个其他答案里第一位的答案。by kerulen

def safe_pawns(pawns):    answer = 0    for pawn in pawns :        if chr(ord(pawn[0])-1)+str(int(pawn[1])-1) in pawns or chr(ord(pawn[0])+1)+str(int(pawn[1])-1) in pawns : answer +=1    return answer

字符串里的数字可以用int转换成整形然后加减。

0 0