一开始还分类重复了。在非0的dfs中居然赋值了0,脑残得一笔。
其实就按 $lead0$ 分类就好了, $lead0$ 表示当前位有没有被赋值过,只有没被赋值过的赋值前导0才会立刻进入下一个数字。
#includeusing namespace std;#define ll long longint cost[]={ 6,2,5,5,4,5,6,3,7,6};int cnt=0;set ,ll> >s;void dfs(int i,int n,ll a,ll b,ll c,int lead0){ if(n<0) return; //cout<<"i="< <<" "<<<" + "<<<" = "< <<" restn= "< < =0) dfs(2,n-cost[0],0,b,c,1); //给a赋值0,那么a就不能搜索了 for(int i=1;i<=9;i++){ //这个bug服气了,非零居然从0开始 if(n-cost[i]>=0){ dfs(1,n-cost[i],i,b,c,0); //给a赋值非0 } } } else{ //a已经被赋值过非0了 //停止搜索a,考虑b,b没被赋值 dfs(2,n,a,b,c,1); for(int i=0;i<=9;i++){ if(n-cost[i]>=0){ dfs(1,n-cost[i],a*10+i,b,c,0); //继续给a加后缀 } } } } else if(i==2){ if(lead0){ //进不来这里,改了之后就可以进来了 dfs(3,n-cost[0],a,0,c,1); //给b赋值0,那么b就不能搜索了 for(int i=1;i<=9;i++){ if(n-cost[i]>=0){ dfs(2,n-cost[i],a,i,c,0); //给b赋值非0 } } } else{ //b已经被赋值过非0了 //停止搜索b,考虑c dfs(3,n,a,b,c,1); for(int i=0;i<=9;i++){ if(n-cost[i]>=0){ dfs(2,n-cost[i],a,b*10+i,c,0); //继续给b加后缀 } } } } else if(i==3){ if(lead0){ //进不来这里? dfs(4,n-cost[0],a,b,0,1); //给c赋值0,那么c就不能搜索了 for(int i=1;i<=9;i++){ if(n-cost[i]>=0){ dfs(3,n-cost[i],a,b,i,0); //给c赋值非0 } } } else{ //停止搜索c dfs(4,n,a,b,c,1); //c已经被赋值过非0了 for(int i=0;i<=9;i++){ if(n-cost[i]>=0){ dfs(3,n-cost[i],a,b,c*10+i,0); //继续给c加后缀 } } } } else{ if(n==0&&(a+b==c)){ s.insert({ {a,b},c}); cnt++; } return; }}int main(){ int n; scanf("%d",&n); n-=4; dfs(1,n,-1,-1,-1,1); printf("%d\n",s.size());}