杭电OJ 2018 母牛的故事

Problem Description

有一头母牛，它每年年初生一头小母牛。每头小母牛从第四个年头开始，每年年初也生一头小母牛。请编程实现在第n年的时候，共有多少头母牛？

 

Input

输入数据由多个测试实例组成，每个测试实例占一行，包括一个整数n(0<n<55)，n的含义如题目中描述。
n=0表示输入数据的结束，不做处理。

 

Output

对于每个测试实例，输出在第n年的时候母牛的数量。
每个输出占一行。

 

Sample Input

  
  

   
   2
4
5
0
  
  

 

Sample Output

  
  

   
   2
4
6
  
  
  
  


  
  

规律： f(n) = f(n-1) + f(n-3)   n>= 4

#include <iostream>
using namespace std;


int main()
{
    int n;
    cin >> n;
    while (n) {
        if (n == 1 || n == 2 || n == 3)
            cout << n << endl;
        else {
            int m1 = 1, m2 = 2, m3 = 3, num = 0;
            for (int i = 4; i <= n; i++) {
                num = m1 + m3;
                m1 = m2;
                m2 = m3;
                m3 = num;
            }
            cout << num << endl;
        }
        cin >> n;
    }
    
}