藍橋杯題目練習 簡單DFS搜索 [藍橋杯]迴文數字 C++實現

[藍橋杯]迴文數字

題目描述

觀察數字：12321，123321 都有一個共同的特徵，無論從左到右讀還是從右向左讀，都是相同的。
這樣的數字叫做：迴文數字。
本題要求你找到一些5位或6位的十進制數字。滿足如下要求：
該數字的各個數位之和等於輸入的整數。

輸入

一個正整數 n (10<n<100), 表示要求滿足的數位和。

輸出

若干行，每行包含一個滿足要求的5位或6位整數。數字按從小到大的順序排列。
如果沒有滿足條件的，輸出：-1

樣例輸入 Copy

45

樣例輸出 Copy

99999

代碼

#include<iostream>
#include<string>
#include<algorithm>
#include<vector>
#include<queue>
#include<cstdio>
#include<cstring>
#include<cmath>
 
using namespace std;
 
typedef long long ll;
const int mod = 999101;
bool vis[10];
int num[7];
int s;
int sum;
bool flag=false,res=false;
void dfs(int index,int n){
	if(sum>s) return;
	if(index==n+1){
	/*
		for(int j=1;j<=n;j++){
			
			cout<<num[j]<<' ';
		}
		cout<<endl;
	*/
		if(sum==s){
			if(!flag){
				flag=true;
			}else{
				cout<<endl;
			}
			for(int j=1;j<=n;j++){
				cout<<num[j];
			}
			res=true;
			
			return;
		}
		return;
	}
	for(int i=0;i<10;i++){
		if(!vis[i] ){
			if(index==1 && i==0) continue;
			if(n==5 && index>3 && i!=num[n-index+1]) continue;
			if(n==6 && index>3 && i!=num[n-index+1]) continue;
			num[index]=i;
			sum+=i;
			dfs(index+1,n);
			sum-=i;
		}
	}
}
int main(){
    int n,m,k;
    cin>>s;
    sum=0;
    dfs(1,5);
    sum=0;
    dfs(1,6);
    if(!res){
    	cout<<-1;
	}
    return 0;
}