CSP 2019-12 回收站选址 python实现

CSP 2019-12 回收站选址 python实现

时间复杂度O(n)

#test_csp_回收站选址
def solve(alist, numlist):
    dic = {
     }
    for i in alist:     #字典一个x轴对应多个y，一对多
        if i[0] in dic:
            dic[i[0]].append(i[1])
        else:
            dic[i[0]] = [i[1]]
    for i in alist:
        count = 0
        try:
            if (i[1]+1 in dic[i[0]]) and (i[1]-1 in dic[i[0]]) and\
               (i[1] in dic[i[0]+1]) and (i[1] in dic[i[0]-1]):     #验证上下左右
                if i[1]+1 in dic[i[0]-1]:   #左上角
                    count += 1
                if i[1]+1 in dic[i[0]+1]:   #右上角
                    count += 1
                if i[1]-1 in dic[i[0]-1]:   #左下角
                    count += 1
                if i[1]-1 in dic[i[0]+1]:   #右下角
                    count += 1
                numlist[count] += 1
        except:
            continue

def main():
    n = eval(input())
    alist = []
    numlist = [0 for _ in range(5)]
    for i in range(n):
        x, y = map(int, input().split(' '))
        alist.append([x, y])
    solve(alist, numlist)
    for i in numlist:
        print(i)

if __name__ == "__main__":
    main()

'''
7
1 2
2 1
0 0
1 1
1 0
2 0
0 1
'''
'''
11
9 10
10 10
11 10
12 10
13 10
11 9
11 8
12 9
10 9
10 11
12 11

'''