自己关于母牛问题的算法

看到了yiding_he 写的《母牛问题（非算法解答） 》感觉手痒痒，也写了一下，使用了一下组合模式，简单测试了一下。

java 代码

1. /**  
2.  *  
3.  */  
4. package cow;   
5.   
6. import java.util.ArrayList;   
7. import java.util.List;   
8.   
9. /**  
10.  * 牛产仔问题。有一头母牛，它每年年初要生一头小母牛；每头小母牛从第四个年头起，  
11.  * 每年年初也要生一头小母牛。按此规律，若无牛死亡，第20年头上共有多少头母牛？  
12.  */  
13. public class CowImp {   
14.     private List chirdenCow = new ArrayList();;   
15.     public int allCows() {   
16.         int count = 1;   
17.         for (int i =0; i < chirdenCow.size(); i++) {   
18.              count = count + ((CowImp)chirdenCow.get(i)).allCows();   
19.         }   
20.         return count;   
21.     }   
22.   
23.     public CowImp(int age) {   
24.         if (age >=  3) {   
25.             for (int i =3; i < age; i++) {   
26.                 chirdenCow.add(new CowImp(age - i));   
27.             }   
28.         }   
29.     }   
30. }   

测试代码，比如四年后，有几头牛。我写的代码假设了那头牛正好3岁了，这样如果测试20年后，传入的值就是23就可以，测试代码如下

 

java 代码

1. package cow;   
2.   
3. import junit.framework.TestCase;   
4.   
5. public class CowImpTest extends TestCase {   
6.   
7.     public void testAllCows() {   
8.         CowImp cow = new CowImp(23);   
9.         assertEquals(2745, cow.allCows());   
10.     }   
11.   
12. }