用Python写，急求解答

Question

Part H ( points)

有一头母牛，它每年年初生一头小母牛。每头小母牛从第四个年头开始，每年年初也生一头小母牛。编写程序，输入一个正整数N，求在第N年的时候，共有多少头母牛？

Answer 1

楼上的写错了
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)