FilesExpand file tree

 master

/

1012.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

32 lines (24 loc) · 556 Bytes

 master

/

1012.py

Copy path

Top

File metadata and controls

• Code
• Blame

32 lines (24 loc) · 556 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

import sys

nums = []

for line in sys.stdin:

nums.append(int(line))

N = nums[0]

K = nums[1]

MAXN = 1000

MAXK = 20

dp = [[0 for y in xrange(MAXK)] for x in xrange(MAXN)]

for i in range(1, K):

dp[0][i] = 1

for i in range(N-1):

for j in range(K):

if dp[i][j] != 0:

if j==0:

for k in range(1, K):

dp[i+1][k] += dp[i][j]

else:

for k in range(K):

dp[i+1][k] += dp[i][j]

my_ans = 0

for i in range(K):

my_ans += dp[N-1][i]

print(my_ans)