蓝桥杯——《123》——python组十二届国赛真题

缱绻命运

已于 2022-05-12 23:25:28 修改

阅读量1.9k

点赞数 7

分类专栏： # 蓝桥杯国赛刷题文章标签：算法

于 2022-05-12 23:24:10 首次发布

本文链接：https://blog.csdn.net/m0_67601373/article/details/124741568

版权

蓝桥杯国赛刷题专栏收录该内容

6 篇文章 2 订阅

订阅专栏

题目描述

小蓝发现了一个有趣的数列，这个数列的前几项如下：

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 行，每行包含一个整数表示对应询问的答案。

样例">样例">样例">样例">样例">样例">样例">样例">样例">输入输出样例

示例

输入

输出

1
4
8

首先，可以把整数划分成第N项第M个数（第几项第几个数）。然后求出前N-1项所有数的和，在加上从1到M个数的和。即前缀和的相关知识。

因为是每一项都是1、2、3、4·····这种顺序所以和则是n*(n+1)//2。即第四项中所有的数的和就是4*（4+1）//2 = 1+2+3+4。

并且第4项的和（10）还是1，1，2，1，2，3，1，2，3，4（数列）第4项的最后一位的排序位置。所以n*(n+1)//2还是第n项最后一个元素的位置。可以借此求出整数在第几项中。

1	1	2	1	2	3	1	2	3	4
1	2	3	4	5	6	7	8	9	10

因为题目的取值范围最大取值是 $10^{12}$ ，由n*(n+1)//2 = $10^{12}$ ，可得最大项为2* $10^{6}$ 。

定义f(N)为第N项的和，sum(N)为前N项的总和：

f(N) = f(N-1) + N；第N项的和，即为n*(n+1)//2。

sum(N) = sum(N-1) + f(N) ；前N项的和

所以只需要分别求出li和ri的前缀和，然后相减即可得出答案。

def F(x):
  return x*(x+1)*(x+2)//6    #前x项的和
def res(x):
  l,r = 1,2000000
  res = 0
  while l<=r:    #判断x在第res+1项，res的值，下面就是二分查找求res的值
    mid = (l+r)//2
    if mid*(mid+1)//2 <= x:
      res = mid
      l = mid+1
    else:
      r = mid-1
  q = x - res*(res+1)//2
  return F(res) + q*(q+1)//2    #q*(q+1)//2是第res项的从1到q的和

T = int(input())
a = []
b = []
for i in range(T):
  l,r = map(int,input().split())    
    #接下来的代码就是分别求出l和r的前缀和，进行相减即可得出答案
  a.append(l-1)
  b.append(r)
for i in range(T):
  ia = res(a[i])
  ib = res(b[i])
  print(ib - ia)