题解 | 特殊城市

import sys

dic = {}
cnt = 0
n = input()
# a[0][:2]不可能是key, 因为不能保证互斥。
# key = 州代码，list放市名
for line in sys.stdin:
    a = line.split()
    if a[0][:2] in dic.keys():
        if a[1] in dic[a[0][:2]] and a[1] != a[0][:2]: #不来自同一个州
            cnt += dic[a[0][:2]].count(a[1])
    if a[1] not in dic.keys():
        dic[a[1]] = []
    if a[1] in dic.keys():
        dic[a[1]].append(a[0][:2])
print(cnt)