《算法笔记-上机实验指南》 DFS

wsfhdhjs

于 2020-10-09 15:49:35 发布

阅读量139

点赞数

CC 4.0 BY-SA版权

分类专栏：《算法笔记上机实验指南》文章标签：算法

本文链接：https://blog.csdn.net/wsfhdhjs/article/details/108980093

《算法笔记上机实验指南》专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一个用于解决特定数学问题的递归算法，该算法通过深度优先搜索来寻找符合要求的整数解，同时利用预计算的幂次表来优化计算过程。文章详细解释了算法的设计思路，并提供了完整的C++实现代码。

A1103

#include<iostream>
#include<vector>
#include<cmath>
using namespace std;

vector<int> pre,temp,fac;

int n,k,p,maxv=-1;

int powl(int x)
{
    int temp=1;
    for(int i=0; i<p; i++)
        temp*=x;
    return temp;
}

void init()                //pre数组中存放的是，某个数的p次方且不超过n
{
    int temp=0,i=0;
    while(temp<=n)
    {
        pre.push_back(temp);
        temp=powl(++i);
    }
}

void DFS(int index,int Nown,int num,int sum)  //DFS：当前访问的数字，已经访问过的个数，所选的数字之和，所选数字的p次方之和
{
    if(Nown==k && sum==n)  //南墙：所选的数字的个数为k   p次方之和刚好为n
    {
        if(num>maxv)        //更新所有数字之和，和对应的数列，注意题中要求数列是递减，所以在主函数中调用DFS时，应该从后往前遍历index
        {
            maxv=num;
            fac=temp;
        }
        return ;
    }
    if(Nown>k || sum>n)   //如果所选数字的个数超过k或者数字之和大于n，则返回
        return ;
    if(index-1>=0)  //注意如果没有这就话，当index为0时，后面还会有index-1，那么对于pre[-1]怎么访问？所以只能是index>=1，也就是index的最小值访问到1
    {
        temp.push_back(index);
        DFS(index,Nown+1,num+index,sum+pre[index]);
        temp.pop_back();
        DFS(index-1,Nown,num,sum);
    }
}

int main()
{
    cin >> n >> k >> p;
    init();    //先找到合适的pre数组
    DFS(pre.size()-1,0,0,0);  //从后向前进行遍历
    if(maxv==-1)      //如果maxv的值没有发生变化，则输入Impossible
        cout << "Impossible";
    else
    {
        printf("%d = %d^%d",n,fac[0],p);
        for(int i=1; i<fac.size(); i++)
        {
            printf(" + %d^%d",fac[i],p);
        }
    }
    return 0;
}