【PAT 乙】1083 是否存在相等的差

题目

给定 N 张卡片，正面分别写上 1、2、……、N，然后全部翻面，洗牌，在背面分别写上 1、2、……、N。将每张牌的正反两面数字相减（大减小），得到 N 个非负差值，其中是否存在相等的差？
输入格式：
输入第一行给出一个正整数 N（2 ≤ N ≤ 10 000），随后一行给出 1 到 N 的一个洗牌后的排列，第 i 个数表示正面写了 i 的那张卡片背面的数字。
输出格式：
按照“差值 重复次数”的格式从大到小输出重复的差值及其重复的次数，每行输出一个结果。
输入样例：
8
3 5 8 6 2 1 4 7
输出样例：
5 2
3 3
2 2

分析

计算差值，计算相同差值的个数，记住是按照从大到小的顺序，我发现pat的题，很多用列表就会超时，但是用字典就不会超时，所以以后解题，能用字典的尽量用字典。

程序

刚开始是用列表做得，一直显示最后一个测试点运行超时。

n=int(input())
num=list(map(int,input().split()))
cha=[]
for i in range(n):
    cha.append(abs(num[i]-i-1))
xu=list(set(cha))
xu=xu[::-1]
for i in xu:
    if cha.count(i)!=1:
        print(i,cha.count(i))

第二种方法就是用字典来做，结果显示全部正确。

n=int(input())
num=list(map(int,input().split()))
cha={}
for i in range(n):
    a=abs(num[i]-i-1)
    if a not in cha:
        cha[a]=1
    else:
        cha[a]+=1
xu=sorted(cha,reverse=True)
for i in xu:
    if cha[i]!=1:
        print(i,cha[i])

知识点