[蓝桥杯 2023 省 A] 平方差
题目描述
给定 L , R L,R L,R,问 L ≤ x ≤ R L \leq x \leq R L≤x≤R 中有多少个数 x x x 满足存在整数 y , z y,z y,z 使得 x = y 2 − z 2 x=y^2-z^2 x=y2−z2。
输入格式
输入一行包含两个整数 L , R L,R L,R,用一个空格分隔。
输出格式
输出一行包含一个整数满足题目给定条件的 x x x 的数量。
样例 #1
样例输入 #1
1 5
样例输出 #1
4
提示
【样例说明】
- 1 = 1 2 − 0 2 1=1^2-0^2 1=12−02
- 3 = 2 2 − 1 2 3=2^2-1^2 3=22−12
- 4 = 2 2 − 0 2 4=2^2-0^2 4=22−02
- 5 = 3 2 − 2 2 5=3^2-2^2 5=32−22
【评测用例规模与约定】
对于 40 % 40 \% 40% 的评测用例, L , R ≤ 5000 L,R \leq 5000 L,R≤5000;
对于所有评测用例, 1 ≤ L ≤ R ≤ 1 0 9 1 \leq L \leq R \leq 10^9 1≤L≤R≤109。
最开始的想法,肯定过不了所有测试点,遍历一些数的平方差 看看是不是在范围内,在的话就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)