zoj - 1204 - Additive equations

先分层，然后进行深度优先遍历。

#include <iostream>#include <algorithm> using namespace std; const int maxn = 30 + 10;       //每组数据最多有30个数int n, ok, C[maxn], a[maxn];        //n为每组数据的个数，ok标记是否有等式存在，C用来存路径，a为输入的数组 bool search(int *a, int x, int y, int v)        //二分搜索，返回搜索成功与否{    int m;    while(x < y)    {        m = x + (y-x)/2;        if(a[m] == v) return 1;        else if(a[m] > v) y = m;        else x = m + 1;    }    return 0;}void dfs(int cur, int des, int cur_sum, int last)       //深度优先遍历，cur为目前层数，des为目标层数，cur_sum为目前数的和,{    int i;    if(cur == des)      //当目前层数==目标层数的时候    {        if(search(a, last, n, cur_sum))     //寻找数组中有没有这个和数存在（为节省时间，从last开始白手搜索）        {            for(i = 0; i < cur-1; i++)      //输出                cout<<C[i]<<"+";            cout<<C[cur-1]<<"="<<cur_sum<<endl;            ok = 1;     //修改全局变量ok，说明有等式存在        }    }    else for(i = last; i < n; i++)      //当目前层数!=目标层数的时候    {        if(cur_sum + a[i] <= a[n-1] && C[cur-1] != a[i])        //第二个比较没有的话数字就可以重复出现了        {            C[cur] = a[i];      //记录路径            dfs(cur+1, des, cur_sum + a[i], i);     //递归调用        }    }} int main(){    int N, i;    cin>>N;     //N组测试数据    while(N--)    {        cin>>n;        for(i = 0; i < n; i++)            cin>>a[i];        sort(a, a+n);       //输完数据后先排序        int sum = 0;        for(i = 0; i < n; i++)      //找最多可以几个数相加，即最大层数        {            sum += a[i];            if(sum >= a[n-1])            break;        }        int len = i+1;      //最大层数        ok = 0;     //注意这里标记为0        for(i = 2; i <= len; i++)       //对每层进行搜索            dfs(0, i, 0, 0);         if(!ok)            cout<<"Can't find any equations."<<endl;        cout<<endl;     //小心，这里的最后一组也要加空行    }    return 0;}