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());
}
05-17 01:34