P1464 Function

题目描述

对于一个递归函数

• 如果 或 或 就返回值 。
• 如果 或 或 就返回
• 如果 并且 就返回 。
• 其它的情况就返回

这是个简单的递归函数，但实现起来可能会有些问题。当 均为 时，调用的次数将非常的多。你要想个办法才行。

注意：例如 又满足条件 又满足条件 ，请按照最上面的条件来算，答案为 。

输入格式

会有若干行。

并以 结束。

输出格式

输出若干行，每一行格式：

w(a, b, c) = ans

注意空格。

输入输出样例 #1

输入 #1

1 1 1
2 2 2
-1 -1 -1

输出 #1

w(1, 1, 1) = 2
w(2, 2, 2) = 4

说明/提示

数据规模与约定

保证输入的数在 之间，并且是整数。

保证不包括 的输入行数 满足 。

解题思路

由递归变递推，记得要开 long long int

AC代码

#include <bits/stdc++.h>

using i64 = long long;
using u64 = unsigned long long;
using u32 = unsigned;
using u128 = unsigned __int128;
#define lowbit(x) = ((x) & -(x))
#define rep_0(a, b, c) for (int a = b; a < c; a++)
#define rep_1(a, b, c) for (int a = b; a <= c; a++)
#define per(a, b, c) for (int a = b; a >= c; a--)
using namespace std;
vector<vector<vector<i64>>> dp(25, vector<vector<i64>>(25, vector<i64>(25, 1)));
void solve()
{
    i64 a, b, c;
    cin >> a >> b >> c;
    for (int i = 1; i <= 20; i++)
    {
        for (int j = 1; j <= 20; j++)
        {
            for (int k = 1; k <= 20; k++)
            {
                if (i < j && j < k)
                {
                    dp[i][j][k] = dp[i][j][k - 1] + dp[i][j - 1][k - 1] - dp[i][j - 1][k];
                }
                else
                {
                    dp[i][j][k] = dp[i - 1][j][k] + dp[i - 1][j - 1][k] + dp[i - 1][j][k - 1] - dp[i - 1][j - 1][k - 1];
                }
            }
        }
    }

    while (a != -1 || b != -1 || c != -1)
    {

        cout << "w(" << a << ", " << b << ", " << c << ") = ";
        if (a < 0 || b < 0 || c < 0)
            cout << "1" << endl;
        else if (a > 20 || b > 20 || c > 20)
            cout << dp[20][20][20] << endl;
        else
            cout << dp[a][b][c] << endl;
        cin >> a >> b >> c;
    }
    return;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int t = 1;
    // cin >> t;

    while (t--)
    {
        solve();
    }

    return 0;
}