母牛-用Python寫，急求解答

用Python寫，急求解答

Part H ( points)

有一頭母牛，它每年年初生一頭小母牛。每頭小母牛從第四個年頭開始，每年年初也生一頭小母牛。編寫程序，輸入一個正整數N，求在第N年的時候，共有多少頭母牛？

最佳回答：

樓上的寫錯了
def row_count(n):
if n==0:
return 1
if n==1:
return 2
if n==2:
return 3
if n==3:
return 4
else:
return row_count(n-3)+row_count(n-1)