算法学习 - 动态规划(DP问题)(C++)

这几天一直再看,觉得看懂了一些,先记下来。

动态规划

动态规划是运筹学的一个方向,就是把多级最优化问题分解成一系列的单阶问题。在不断增加的过程中,不断的计算当前问题的最优解。

一般分为如下四个部分:

  • 线性动规:拦截导弹,合唱队形,挖地雷,建学校,剑客决斗等;
  • 区域动规:石子合并, 加分二叉树,统计单词个数,炮兵布阵等;
  • 树形动规:贪吃的九头龙,二分查找树,聚会的欢乐,数字三角形等;
  • 背包问题:01背包问题,完全背包问题,分组背包问题,二维背包,装箱问题,挤牛奶(同济ACM第1132题)等;

汽车生产线问题

这个问题是《算法导论》的动态规划的例题,我自己觉得这道题比较简单而且典型,所以就解释下这个题目:

Colonel汽车公司在有两条装配线的工厂里生成汽车。每一条装配线上有n个装配站,
两条生产线上相同位置的装配站功能相同,但所需时间不同,并且汽车底盘在两条
装配线间转移要花费一定的时间。如下图所示两条生产线。
技术分享

首先我们知道每个阶段的最短时间,都包含了上一阶段的最短时间。

而比较简单的是,我们只有两种情况,一种是在装配线1上时间短,一种是在装配线2上时间短。

假如暴力搜索,贪心去算得话(也就是递归的办法)。时间复杂就是2^n,这个是不可接受的。

这个时候我们的办法是先从第一个问题开始,装配线1, 2上只有一个station。那么比较简单,一比较就好了。当有两个station的时候,就是从上一次比较的结果中(一个line 1最短,一个line 2最短)中继续加上当前station的值继续比较。

所以我们发现是每个当前的最优解,都包含了上一次的最优解。

代码

代码就是我们从最开始,一直保存当前问题的最优解,然后去求的下一次的最优解。

//
//  main.cpp
//  DP_line
//
//  Created by Alps on 15/4/26.
//  Copyright (c) 2015年 chen. All rights reserved.
//

#include <iostream>
using namespace std;

#define NUM 5

int main(){
    int first[NUM];//到装备站1,i的最短路径长度
    int second[NUM];//到装配站2,i的最短路径长度
    int linef[NUM]; //从1进入的路径
    int lines[NUM]; //从2进入的路径
    int a[NUM]; //在装配站1,i 所需要呆的时间
    int b[NUM]; //再装配站2,i 所需要呆的时间
    int m[NUM]; //从装配站1,i-1 到装配站2,i的时间
    int n[NUM]; //从装配站2,i-1 到装配站1,i的时间
    int line[NUM]; //当前最短路经所经过的装配站
    int f[NUM]; //当前最短路径长度
    int ea,eb,xa,xb; // ea为进入装配站1时间,eb为进入2的时间,xa为出装配站1的时间,xb为出装配站2的

    scanf("%d %d %d %d",&ea,&eb,&xa,&xb);
    for(int i=0;i<NUM;++i)
    {
        scanf("%d %d", &a[i], &b[i]);
    }
    first[0] = ea + a[0];
    second[0] = eb + b[0];
    for(int i=0;i<NUM-1;++i)
    {
        scanf("%d %d", &m[i], &n[i]);
    }
    for(int i=1;i<NUM;++i)
    {
        if(first[i-1] + a[i] < second[i-1] + m[i-1] + a[i])
        {
            first[i] = first[i-1] + a[i];
            linef[i] = 1;
        }else{
            first[i] = second[i-1] + m[i-1] + a[i];
            linef[i] = 2;
        }
        if(second[i-1] + b[i] < first[i-1] + n[i-1] + b[i])
        {
            second[i] = second[i-1] + b[i];
            lines[i] = 2;
        }else
        {
            second[i] = first[i-1] + n[i-1] + b[i-1];
            lines[i] = 1;
        }
    }
    for(int i=0;i<NUM;++i)
    {
        if(first[i] + xa < second[i] + xb)
        {
            f[i] = first[i] + xa;
            line[i] = 1;
        }else{
            f[i] = second[i] + xb;
            line[i] = 2;
        }
    }

    for(int i=0;i<NUM;++i)
    {
        printf("station %d\n",line[i]);
    }
    printf("Distince is %d\n",f[NUM-1]);

    return 0;
}

这个代码比较简单,精华就是那个for循环了。

测试用例如下:

3 2 3 4
4 3
3 6
6 3
2 3
5 2
2 3
2 4
3 4
4 3

假如有任何建议,欢迎评论。互相学习。谢谢。

郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。