不定方程x1+x2+x3=5的非负整数解的个数、空间分割

不定方程x1+x2+x3=5的非负整数解的个数为:
1.三个不同的数{4,1,0}，{3,2,0}。每个有A33的排列共62=12种；
2.三个数中两个相同的{5,0,0}，{3,1,1}，{2,1,2}。。每个有A13种排列共33=9种 和为21种

空间分割：线分平面：[n(n+1)]/2+1
平面分空间：(n^3+5n+6)/6

Python读取键盘输入：n = int(input())
nums = list(map(int, input().split()))
ks = int(input())