[dfs] 洛谷 P1149

题目

题目链接：https://www.luogu.com.cn/problem/P1149

思路

dfs一个一个数匹配看看能不能满足就好了。但要注意处理二位数三位数。

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ll long long
using namespace std;
const int INF = 0x3f3f3f3f;
int x[1001] = {6, 2, 5, 5, 4, 5, 6, 3, 7, 6}, b[4];
int res=0,n;
void dfs(int l){
	for(int i=0;i<=999;i++){
		if(n>=x[i]){
			b[l]=i;
			n=n-x[i];
			if(l==3) {
				if(n==0&&b[1]+b[2]==b[3]) res++;
 			}
			else dfs(l+1);
			n=n+x[i];
		}
	}
}
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	cin>>n;
	n-=4;
	for(int i=10;i<=999;i++){
		x[i]=x[i%10]+x[i/10];
	}
	dfs(1);
	cout<<res<<endl;
    return 0;
}