（水题）洛谷 - P1149 - 火柴棒等式

https://www.luogu.org/problemnew/show/P1149

一开始还分类重复了。在非0的dfs中居然赋值了0，脑残得一笔。

其实就按 $lead0$ 分类就好了， $lead0$ 表示当前位有没有被赋值过，只有没被赋值过的赋值前导0才会立刻进入下一个数字。

#include<bits/stdc++.h>
using namespace std;
#define ll long long

int cost[]={,,,,,,,,,};

int cnt=;

set<pair<pair<ll,ll>,ll> >s;

void dfs(int i,int n,ll a,ll b,ll c,int lead0){
    if(n<)
        return;
    //cout<<"i="<<i<<" "<<a<<" + "<<b<<" = "<<c<<" restn= "<<n <<endl;
    //lead0=1，表示这个数可以取前导0，也就是这个数还没有被赋值
    if(i==){
        if(lead0){
            if(n-cost[]>=)
                dfs(,n-cost[],,b,c,);
                //给a赋值0，那么a就不能搜索了
            for(int i=;i<=;i++){
                //这个bug服气了，非零居然从0开始
                if(n-cost[i]>=){
                    dfs(,n-cost[i],i,b,c,);
                    //给a赋值非0
                }
            }
        }
        else{
            //a已经被赋值过非0了

            //停止搜索a，考虑b，b没被赋值
            dfs(,n,a,b,c,);

            for(int i=;i<=;i++){
                if(n-cost[i]>=){
                    dfs(,n-cost[i],a*+i,b,c,);
                    //继续给a加后缀
                }
            }
        }
    }
    else if(i==){
        if(lead0){
            //进不来这里，改了之后就可以进来了
            dfs(,n-cost[],a,,c,);
            //给b赋值0，那么b就不能搜索了
            for(int i=;i<=;i++){
                if(n-cost[i]>=){
                    dfs(,n-cost[i],a,i,c,);
                    //给b赋值非0
                }
            }
        }
        else{
            //b已经被赋值过非0了

            //停止搜索b，考虑c
            dfs(,n,a,b,c,);

            for(int i=;i<=;i++){
                if(n-cost[i]>=){
                    dfs(,n-cost[i],a,b*+i,c,);
                    //继续给b加后缀
                }
            }
        }
    }
    else if(i==){
        if(lead0){
            //进不来这里?
            dfs(,n-cost[],a,b,,);
            //给c赋值0，那么c就不能搜索了
            for(int i=;i<=;i++){
                if(n-cost[i]>=){
                    dfs(,n-cost[i],a,b,i,);
                    //给c赋值非0
                }
            }
        }
        else{
            //停止搜索c
            dfs(,n,a,b,c,);
            //c已经被赋值过非0了
            for(int i=;i<=;i++){
                if(n-cost[i]>=){
                    dfs(,n-cost[i],a,b,c*+i,);
                    //继续给c加后缀
                }
            }
        }
    }
    else{
        if(n==&&(a+b==c)){
            s.insert({{a,b},c});
            cnt++;
        }
        return;
    }
}

int main(){
    int n;
    scanf("%d",&n);
    n-=;
    dfs(,n,-,-,-,);
    printf("%d\n",s.size());
}