P9231 [蓝桥杯 2023 省 A] 平方差 Python 打表+前缀和

最新推荐文章于 2024-08-19 17:12:46 发布

太阳好刺眼

最新推荐文章于 2024-08-19 17:12:46 发布

阅读量1.0k

点赞数 22

文章标签：蓝桥杯 python 职场和发展

本文链接：https://blog.csdn.net/weixin_43843099/article/details/137588104

版权

该文介绍了解决编程竞赛题目中关于区间内满足特定平方差关系的数计数问题，涉及算法优化和数学技巧。

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

[蓝桥杯 2023 省 A] 平方差

题目描述

给定 $L, R$ ，问 $\leq x \leq R$ 中有多少个数 $x$ 满足存在整数 $y, z$ 使得 $x=y^2-z^2$ 。

输入格式

输入一行包含两个整数 $L, R$ ，用一个空格分隔。

输出格式

输出一行包含一个整数满足题目给定条件的 $x$ 的数量。

样例 #1

样例输入 #1

1 5

样例输出 #1

提示

【样例说明】

$1=1^2-0^2$
$3=2^2-1^2$
$4=2^2-0^2$
$5=3^2-2^2$

【评测用例规模与约定】

对于 $\%$ 的评测用例， $\leq 5000$ ；

对于所有评测用例， $\leq L \leq R \leq 10^9$ 。

最开始的想法，肯定过不了所有测试点，遍历一些数的平方差看看是不是在范围内，在的话就count+1，但是这个思路，一个测试点都不能通过。前四个时WA，后面时MLE ，居然一个点都不能通过。

l,r = map(int,input().split())
s = [i for i in range(l,r+1)]
count = 0
for i in range(50):
    for j in range(i):
        if i**2 - j**2 in s:
            count +=1
print(count)

后面看了看别人的思路，发现这个x值是有规律的，x是奇数或者x是4的倍数都是满足条件的，所以题目变成了统计给定范围内的奇数个数和4的倍数的个数，但就算简化了问题，遍历统计的话还是会超时。

后面看了b站一位博主的思路，用前缀和的方法解决。
即（l,r）= (1,r) - (1,l-1)

满足条件的x的输出，看规律
1 3 4 5 7 8 9 9 11 12 13 15 15 16 16 17 19 20 21 21 23 24 24 25 25 27 27 28 29 31 32 32 33 33 35 35 36 36 37 39 39 40 40 41 43 44 45 45 45 47 48 48 48 49 49

答案

import math
l,r = map(int,input().split())

sum = 0
count_ji_r = math.ceil(r/2)
count_ji_l = math.ceil((l-1)/2)

count_four_r = r//4
count_four_l = (l-1)//4

sum = count_four_r - count_four_l + count_ji_r - count_ji_l
print(sum)