Python: найдите количество комбинаций

Python Basic - 1: Упражнение 62 с решением

Напишите программу на Python, чтобы найти число комбинаций, которые удовлетворяют p + q + r + s = n, где n - данное число <= 4000, а p, q, r, s - в диапазоне от 0 до 1000.

Пример решения :

Код Python:

from collections import Counter
print("Input a positive integer: (ctrl+d to exit)") 
pair_dict = Counter()
for i in range(2001):
  pair_dict[i] = min(i, 2000 - i) + 1 
 
while True:
  try:
    n = int(input())
    ans = 0
    for i in range(n + 1):
      ans += pair_dict[i] * pair_dict[n - i]
    print("Number of combinations of a,b,c,d:",ans) 
  except EOFError:
    break

Пример вывода:

 Введите положительное целое число: (Ctrl + D для выхода)
 252
Количество комбинаций a, b, c, d: 2731135

Блок - схема:

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Напишите программу на Python, которая вычисляет максимальное значение суммы передаваемых целых чисел.
Далее: Напишите программу на Python, которая суммирует столбцы и строки данной таблицы, как показано на указанном рисунке.