第五届CCF PTA Python编程培训师资认证（P试卷）

第一题 卫星发射

试题名称：卫星发射
时间限制： 1.0s
内存限制： 128.0MB
【问题描述】
在 2050 年卫星发射技术已经得到极大发展，我国将援助 A 国建立远轨道卫星导航系统，该项目计划第一个天发射一颗卫星；之后两天（第二天和第三天），每天发射两颗卫星；之后三天（第四、五、六天），每天发射三颗卫星；之后四天（第七、八、九、十天），每天发射四颗卫星……；这种卫星发射模式会一直延续到项目完成：当连续 n 天每天发射 n 颗卫星后，航天局在之后的连续 n+1 天里，每天发射 n+1 颗卫星。
请计算在前 k 天里，一共发射了多少卫星。
【输入描述】
一个正整数 k，表示发射卫星的天数。
【输出描述】
一个正整数，即发射卫星的总数。
【输入样例】
6
【输出样例】
14
【样例 1 说明】
航天局第一天发射一颗卫星；第二天和第三天，每天发射两颗卫星；第四、五、六天，每天发射三颗卫星。因此一共发射 1+2+2+3+3+3=14 颗卫星。
【数据规模】
对于 100%的数据， 1≤k≤10^4。

k=int(input())
n=1
wx=0
while n<=k:
    wx+=n*n
    k-=n
    n+=1
if k!=0:
    wx+=n*k
print(wx)

第二题 伯罗奔尼撒箭阵

试题名称：伯罗奔尼撒箭阵
时间限制： 1.0s
内存限制： 128.0MB
【问题描述】
在伯罗奔尼撒战争中，为了应对敌方的箭阵，指挥官正在研究一种新的列队方式，为了方便士兵理解，抽象如下：给出正整数 n，要求按如下方式构造数列：

1. 只有一个数字 n 的数列是一个合法的数列。
2. 在一个合法的数列的末尾加入一个正整数，但是这个正整数不能超过该数列最后一项的一半，可以得到一个新的合法数列。
   请你帮助指挥官求出，一共有多少个合法的数列。两个合法数列 a, b 不同当且仅当两数列长度不同或
   存在一个正整数 i≤∣a∣，使得 ai ≠bi。
   【输入描述】
   输入只有一行一个整数，表示 n。
   【输出描述】
   输出一行一个整数，表示合法的数列个数。
   【输入样例】
   6
   【输出样例】
   6
   样例 1 解释
   满足条件的数列为：
   6
   6,1
   6,2
   6,3
   6,2,1
   6,3,1
   【数据规模】
   数据保证，对于全部的测试点，保证 1≤n≤10^3。

def f(n):
    global count
    b=n//2
    count+=b
    for i in range(<