排列小球（蓝桥杯）

题目描述

小蓝有黄绿蓝三种颜色的小球，分别为 R, G, BR,G,B 个。同样颜色的小球没有区别。

小蓝将这些小球从左到右排成一排，排完后，将最左边的连续同色小球个数记为 t_1t1​，将接下来的连续小球个数记为 t_2t2​，以此类推直到最右边的小球。

请问，总共有多少总摆放小球的方案，使得 t1, t2, ⋯ 为严格单调递增序列，即 t1​≤t2​≤t3​≤⋯。

输入描述

输入一行包含三个整数 R, G, BR,G,B。

其中，0≤R,G,B≤50。

输出描述

输出一个整数，表示答案。

输入输出样例

示例 1

输入

3 6 0

输出

3

样例说明

用 r 表示红球，g 表示绿球，可能的方案包括：

rrrgggggg

grrrggggg

ggrrrgggg

运行限制

• 最大运行时间：1s
• 最大运行内存: 128M

思路：直接模拟暴力搜索就行，但是注意去重

最终判断排成的序列是否用完所有的小球就行

AC代码

#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<set>
#include<cmath> 
#include<cstdio>
#include<vector> 
#include<stack>
#include<queue>
#include<algorithm>
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define PII pair<int,int>
#define inf 0x3f3f3f3f
#define endl "\n"
typedef long long ll;
using namespace std;
int t,n,m,k;
string s;
map<string,int> mp;
int ans;
void dfs(int a,int b,int c,int next,string str)
{
    if(!a && !b && !c){
        if(!mp[str]){
            mp[str] = 1;
            ans++;
        }
        return ;
    }
    if(a > next){
        for(int i = next + 1 ; i <= a ; i++){
            dfs(a - i,b,c,i,str += i);
        }
    }
    if(b > next){
        for(int i = next + 1 ; i <= b ; i++){
            dfs(a,b - i,c,i,str += i);
        }
    }
    if(c > next){
        for(int i = next + 1 ; i <= c ; i++){
            dfs(a,b,c - i,i,str += i);
        }
    }
}
void solve()
{
    int a,b,c;
    cin >> a >> b >> c;
    dfs(a,b,c,0,"");
    cout << ans << endl;
}
int main() {
    IOS;                           
    solve();     
    return 0; 
}