洛谷_P1102 A-B 数对_python写法

最新推荐文章于 2024-10-30 13:16:11 发布

Hooray11

最新推荐文章于 2024-10-30 13:16:11 发布

阅读量386

点赞数 3

分类专栏：刷题笔记文章标签： python 算法开发语言

本文链接：https://blog.csdn.net/Hooray11/article/details/136971418

版权

刷题笔记专栏收录该内容

50 篇文章 11 订阅

订阅专栏

本文讨论了解决洛谷竞赛中P1102A-B数对问题的不同方法，首先介绍了使用列表和字典的解法，然后重点转向Python内置bisect库在二分查找中的应用，展示了代码优化的过程。

摘要由CSDN通过智能技术生成

P1102 A-B 数对 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

解法一：

n, c = map(int,input().split())

data = list(map(int,input().split()))
data.sort()
maxx = max(data)

sum = 0
l = [0 for _ in range(maxx+1)]
for i in data:
    l[i] += 1
for b in range(1, maxx+1):
    if b+c <= maxx and l[b] != 0:
        num = l[b+c] * l[b]
        sum += num

print(sum)

一开始用的列表，但是有一个案例无法通过，现在就改成用字典了。

n, c = map(int,input().split())

data = list(map(int,input().split()))

sum = 0
dict = {}
for i in data:
    if i not in dict.keys():
        dict[i] = 1
    else:
        dict[i] += 1
for i in data:
    if i+c in dict.keys():
        sum += dict[i+c]
print(sum)