题目背景
欢迎新斗友
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。