题目背景
欢迎新斗友
https://www.luogu.com.cn/user/358957
题目描述
定义一个长度为 m 的非空序列 p1,p2,...,pm 是乱的当且仅当满足以下两个条件。
- 所有元素之和不超过 n,即 ∑i=1mpi≤n。
- 对于任意一个元素 pi 满足 pi=1 或 pi 为质数。
定义一个乱的序列 p1,p2,...,pm 的乱斗值为该序列中所有元素减 k 的平方和,即 ∑i=1m(pi−k)2。
特别的,定义一个不乱的序列的乱斗值为 0。
现在给定两个正整数 n,k,问所有序列中乱斗值最大的序列的乱斗值是多少。
输入格式
本题有多组测试数据。第一行输入一个正整数 T,表示测试数据组数,接下来分别输入 T 组数据。
对于每组测试数据,输入一行两个正整数 n,k。
输出格式
对于每组测试数据输出一行一个整数表示答案。
5
1 1
2 1
4 1
5 2
10 10
0
1
4
9
810
提示
样例解释
对于样例第 1,2,3,4 组数据,其中一种乱斗值最大的序列分别为 (1),(2),(1,3),(5)。
数据范围与约定
| 测试点编号 | T | n | k | 特殊性质 | 
| 1 | =100 | ≤10 | ≤10 | 无 | 
| 2 | =200 | ≤30 | 
| 3 | =300 | ≤103 | ≤5×104 | 
| 4 | =400 | ≤105 | 
| 5 | =500 | ≤107 | 
| 6 | =600 | ≤109 | =1 | n 为质数 | 
| 7 | =700 | 无 | 
| 8 | =800 | =44444 | 
| 9 | =900 | ≤5×104 | n 为质数 | 
| 10 | =103 | 无 | 
对于所有测试点,保证 1≤T≤103,1≤n≤109,1≤k≤5×104。