一道面試題。1x+2y+5z=n

寫一個函數int foo(int n)，對於給定的n，求出對於任意{x,y,z}>=0，1x+2y+5z=n，組合數

#include<iostream>

using namespace std;

int foo1(int n)
{
    int count=0;
    for(int z =0 ; z <= n/5; z++)
    {
            for(int y = 0; y <= (n - 5*z)/2; y++)
            {
                    for(int x = 0; x <= (n-5*z-2*y); x++)
                    {
                            count++;
                    }// 這個迴圈可以替換成count += n-5*z-2*y + 1
            }
    }
    return count;
}

int foo2(int n)
{
    int count = 0;
    for(int z = 0; z <= n/5; z++)
    {
            for(int y = 0; y <= (n - 5*z)/2; y++)
            {
                    count += n-5*z-2*y + 1;
            }//這個迴圈可以替換，y是遞增數列
    }
    return count;
}

int foo3(int n)
{
    int count = 0;
    for(int z = 0; z <= n/5; z++)
    {
         count += (n - 5*z + 2)*(n - 5*z + 2)/4;
    }
    return count;
}

int main()
{
    cout << foo1(10) << endl;
    cout << foo2(10) << endl;
    cout << foo3(10) << endl;
    system("pause");
    return 0;
}