想要了解其它题目请点击http://t.csdn.cn/U2r3g
小蓝发现了一个有趣的数列,这个数列的前几项如下:
1,1,2,1,2,3,1,2,3,4,⋯
小蓝发现,这个数列前 1 项是整数 1,接下来 2 项是整数 1 至 2,接下来 3 项是整数 1 至 3,接下来 4 项是整数 1 至 4,依次类推。
小蓝想知道,这个数列中,连续一段的和是多少。
输入描述
输入的第一行包含一个整数 T,表示询问的个数。
接下来 T 行,每行包含一组询问,其中第 i 行包含两个整数 li 和 ri,表示询问数列中第 li 个数到第 ri 个数的和。
输出描述
输出 T 行,每行包含一个整数表示对应询问的答案。
输入输出样例
示例
输入
3
1 1
1 3
5 8
输出
1
4
8
评测用例规模与约定
对于 10% 的评测用例,1≤T≤30,1≤li≤ri≤100。
对于 20% 的评测用例,1≤T≤100,1≤li≤ri≤1000。
对于 40% 的评测用例,1≤T≤1000,1≤li≤ri≤1e6。
对于 70% 的评测用例,1≤T≤10000,1≤li≤ri≤1e9。
对于 80% 的评测用例,1≤T≤1000,1≤li≤ri≤1e12。
对于 90% 的评测用例,1≤T≤10000,1≤li≤ri≤1e12。
对于所有评测用例,1≤T≤100000,1≤li≤ri≤1e12。
运行限制
- 最大运行时间:5s
- 最大运行内存: 256M
思路:
首先这个题肯定是要用到前缀和的,而前缀和数组的长度我们需要先确定一下
##初始化
# sum=[0,1]#元素数量前缀和
# i=1
# while True:#最后一项小于
# i += 1
# sum.append(i+sum[-1])
# if sum[-1]>int(1e12):
# break
#len(sum)==1414215
#以上求出 最大长度1414215
之后我们建立两个前缀和数组,一个是关于‘个数’(sum) 一个是关于‘和’sum1,其中的每一项代表一个轮回区间,例如:【1,1,2,1,2,3】就有三个轮回区间
例如:【1,1,2,1,2,3 】,
sum:【1,3, 6】,解释:每一项代表从第一个轮回区间到当前轮回区间的数字个数
sum1:【1,4,10】,解释:每一项代表从第一个轮回区间到当前轮回区间的和
对于给定的范围 ‘l’与‘r’,我们用二分分别找出其在sum的位置,也就是其所在第几个轮回区间,也就是找出sum中第一个大于它的元素
例如:l=2,r=5。l在第一个元素中,而r在第三个元素中
由于我们找到的l和r在各自的轮回区间中可能处于任何位置,所以我们要求出l所在区间的 l~区间结尾 的和 与r所在区间的 区间开头~r 的和,如果说l所在区间与r所在区间之间有别的区间的话,就求出他们之间相隔的区间的和
例如:【1,1,2,1,2,3,1,2,3,4,1,2,3,4,5】 l=2, r=14, 我们先求出l所在区间的以l开头的和为:3 ,求出r所在区间的以r结尾的和为:10,再求出l所在区间与r所在区间相隔的区间的和为:16
对于求l/r所在区间的和,我们可以用等差数列或者sum,对于求多个连续区间的和,我们会用到sum1
代码实现:
##用二分找出l和r在sum的什么位置,也就是找从左往右第一个大于
def erf(x):
global sum,L
l,r=0,L-1
while l<r:
mid=(l+r)//2
if sum[mid]<x:
l=mid+1
else:
r=mid
return l
def arith(x):
return x*(1+x)//2
##初始化
# sum=[0,1]#元素数量前缀和
# i=1
# while True:#最后一项小于
# i += 1
# sum.append(i+sum[-1])
# if sum[-1]>int(1e12):
# break
#len(sum)==1414215
#以上求出 最大长度1414215
L=1414217
sum=[0 for i in range(L)]
sum1=[0 for i in range(L)]
for i in range(1,L):
sum[i]+=sum[i-1]+i
sum1[i]+=sum1[i-1]+sum[i]
T=int(input())
for _ in range(T):
l,r=map(int,input().split())
lw=erf(l)#an
rw=erf(r)
ldi=l-sum[lw-1]#l在该区间第几个,,a1
rdi=r-sum[rw-1]#r在该区间第几个
total=0#最终区间和
if lw==rw:
# total+=((rdi-ldi+1)*(ldi+rdi))//2
total+=sum[rdi]-sum[ldi-1]
else:
# lshen=((lw-ldi+1)*(ldi+lw))//2#l所在区间还剩的
# rshen=((rdi)*(1+rdi))//2#r所在区间还剩的
# total+=lshen+rshen
total+=sum[rdi]+(sum[lw]-sum[ldi-1])
if lw<rw-1:#中间至少隔了一个
total+=sum1[rw-1]-sum1[lw]
print(total)