回答編集履歴
1
answer
CHANGED
|
@@ -95,8 +95,10 @@
|
|
|
95
95
|
|
|
96
96
|
N, K = map(int, input().split())
|
|
97
97
|
n = comb(K - 1, 2)
|
|
98
|
-
if (m := K - N - 1) >= 2:
|
|
98
|
+
if (m := K - N - 1) >= 2: n -= 3 * comb(m, 2)
|
|
99
|
-
if (m := K - 2 * N - 1) >= 2:
|
|
99
|
+
if (m := K - 2 * N - 1) >= 2:
|
|
100
|
+
n += 3 * comb(m, 2)
|
|
100
|
-
if (m := K - 3 * N - 1) >= 2:
|
|
101
|
+
if (m := K - 3 * N - 1) >= 2:
|
|
102
|
+
n -= comb(m, 2)
|
|
101
103
|
print(n)
|
|
102
104
|
```
|