“深圳计算科研院杯“E起来编程暨第三届湖北省赛 C Hamster and Equation

题意： 给两个数n ，k ，在【-n，n】的范围内，满足以上公式。问一共有几组例子。
思路：
a=x1(x1+1)
将公式化简为a+b=k(c+d)；
以例子1为例子，a的值只能取到0，2.
x1=0，-1时a是0，x1=1时，a=2.
然后，根据化简后的公式，可以得到

a b c d
0 0 0 0
0 2 0 2
0 2 2 0
2 0 2 0
2 0 0 2
2 2 2 2

这六种情况，然后 是排列组合的思路：
0 0 0 0 ==>2 * 2 * 2 * 2 =16
就是0有两种情况
那么总：
16+4+4+4+4+1=33

understand?

for _ in range(int(input())):
    n,k=map(int,input().split())
    x=[]
    for i in range(-n,n+1):
        x.append(i*i+i)
    # print(x)
    l={}
    for x1 in x:
        for x2 in x:
            tem=x1+x2
            if tem in l:
                l[tem]+=1
            else:l[tem]=1
    # print(l)
    res=0
    for j in l:
        if j*k in l:
            res+=l[j]*l[j*k]
    print(res)