入门OJ 6247【算式等式】

描述

　　给你N个数字，每个数字均不相同，你可以取其中任意个数字相加，如果所得到的结果在给出的数列中也能找到，则输出这个等式

输入输出格式

输入

　　第一行一个数字N
　　接下来N个数字

输出

　　输出你找到的等式，注意看样例的输出格式 

输入输出样例

输入样例1

6
1 3 5 7 8 9

输出样例1

1+7=8
1+8=9
3+5=8
1+3+5=9

解题思路

　　这道题首先想到的就是分支限界法（<--不知道的童鞋赶紧去看看）限制了深度后才让输出的数字数量从少到多。

题解

#include<bits/stdc++.h>
using namespace std;
int n,q;
int num[101];
set<int> s;//方便寻找 
bool flag[10001];//标记 
bool qwe=false;
void dfs(int sum,int pre,int dep)//总值，开始搜索的下标，深度 
{
    if(dep==q+1)//深度超出 
    {
        if(s.count(sum))//找得到 
        {
            qwe=true;//控制格式 
            for(int i=1;i<=n;i++)
            {
                if(flag[i])//有这个数就输出 
                {
                    if(!qwe)cout<<"+"<<num[i];
                    else
                    {
                        cout<<num[i];
                        qwe=false;
                    }
                }
                  
            }
            cout<<"="<<sum<<endl;
        }
        return;//注意，这这里无论找不找得到都要退出返回 
    }
    for(int i=pre;i<=n;i++)
    {
        if(!flag[i])
        {
            flag[i]=true;
            dfs(sum+num[i],i+1,dep+1);
            flag[i]=false;//取消标记，回溯操作 
        }
    }
}
int main()
{
   cin>>n;
   for(int i=1;i<=n;i++)
   {
        cin>>num[i];
        s.insert(num[i]);//存入set 
   }
   for( q=2;q<n;q++)//分支限界精髓重点，循环深度搜索，注意全局变量 
   {
        dfs(0,1,1);
   }
    return 0;
}