选取一组不相邻的数字使其和最大

最新推荐文章于 2023-12-09 16:01:42 发布

Chaoying.

最新推荐文章于 2023-12-09 16:01:42 发布

阅读量245

点赞数

分类专栏：动态规划文章标签：算法 c++ 开发语言动态规划

本文链接：https://blog.csdn.net/TJLCY/article/details/129728011

版权

动态规划专栏收录该内容

1 篇文章 0 订阅

订阅专栏

该代码实现了一个算法，从给定的一组数字中选择互不相邻的数字，以使它们的和最大化。通过动态规划建立f(i)的递推关系，即f(i)=max(f(i-2)+v(i),f(i-1))，并存储选择方案。最后，给出最大和及对应的数字下标。

摘要由CSDN通过智能技术生成

参考教程

/*
问题描述：
有一组数字如4 1 1 9 1,现要求从中随机选取一组互不相邻的数字使得选取的数字之和最大
分析：
优化目标： max(sum)
约束：不相邻

f(i)表示从前i个数中选取得到的最大值，则有如下递推关系
f(i)=f(i-2)+v(i)  选第i个数
f(i)=f(i-1) 不选第i个数
f(1)=v(1)
f(2)=max(v(1),v(2))
*/

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

// SSM:sum selected max
void SSM(const vector<int>s,int & smax,vector<int> &index)
{
    int n = s.size();
    // 最大和
    vector<int> res(n,0);
    // 选择方案
    vector<vector<int>> indexs(n,vector<int>());
    res[0] = s[0];
    indexs[0].push_back(0);
    if(n==1)
    {
        index.push_back(0);
        smax = s[0];
        return;
    }
    if(s[0]>=s[1])
    {
        res[1] = s[0];
        indexs[1].push_back(0);
        index.push_back(0);
        smax = s[0];
    }
    else
    {
        res[1] = s[1];
        indexs[1].push_back(1);
        index.push_back(1);
        smax = s[1];
    }
    if(n==2)
        return;
    for (int i = 2; i < n;i++)
    {
        int r1 = s[i] + res[i - 2];
        int r2 = res[i - 1];
        if(r1>=r2)
        {
            res[i] = r1;
            indexs[i]=vector<int>(indexs[i-2]);
            indexs[i].push_back(i);
        }
        else
        {
            res[i] = r2;
            indexs[i] = vector<int>(indexs[i - 1]);
        }
    }
    index = indexs[n - 1];
    smax = res[n - 1];
    return;
}

int main()
{
    vector<int> s;
    s.push_back(1);
    s.push_back(2);
    s.push_back(4);
    s.push_back(1);
    s.push_back(7);
    s.push_back(8);
    s.push_back(3);
    int res;
    vector<int> index;
    SSM(s,res,index);
    cout << "最大值为:" << res << endl;
    cout<<"选取的元素下标为:"<<endl;
    for (vector<int>::iterator it = index.begin(); it != index.end();it++)
    {
        cout << *it << "\t";
    }
    cout << endl;
    system("pause");
    return 0;
}