【大疆2019秋招在线笔试】算法第一题参考思路
题目
给定暑假时间X天(<=1000),游戏数量N个(<=11),接下来N行给定每种游戏需要花费的天数(Ai),以及通关该游戏带来的成就点数(Bi),求:在暑假X天里能够达成的最高成就点数。
总体思路
首先对游戏进行排序(时间从短到长,价值从低到高),然后就是标准的01背包问题了。
背包问题的解决过程
在解决问题之前,为描述方便,首先定义一些变量:Vi表示第 i 个物品的价值,Wi表示第 i 个物品的体积,定义V(i,j):当前背包容量 j,前 i 个物品最佳组合对应的价值,同时背包问题抽象化(X1,X2,…,Xn,其中 Xi 取0或1,表示第 i 个物品选或不选)。
1、建立模型,即求max(V1X1+V2X2+…+VnXn);
2、寻找约束条件,W1X1+W2X2+…+WnXn<capacity;
3、寻找递推关系式,面对当前商品有两种可能性:
包的容量比该商品体积小,装不下,此时的价值与前i-1个的价值是一样的,即V(i,j)=V(i-1,j);
还有足够的容量可以装该商品,但装了也不一定达到当前最优价值,所以在装与不装之间选择最优的一个,即V(i,j)=max{V(i-1,j),V(i-1,j-w(i))+v(i)}。
其中V(i-1,j)表示不装,V(i-1,j-w(i))+v(i) 表示装了第i个商品,背包容量减少w(i),但价值增加了v(i);由此可以得出递推关系式:
j<w(i) V(i,j)=V(i-1,j)
j>=w(i) V(i,j)=max{V(i-1,j),V(i-1,j-w(i))+v(i)}
这里需要解释一下,为什么能装的情况下,需要这样求解(这才是本问题的关键所在!):可以这么理解,如果要到达V(i,j)这一个状态有几种方式?
肯定是两种,第一种是第i件商品没有装进去,第二种是第i件商品装进去了。没有装进去很好理解,就是V(i-1,j);装进去了怎么理解呢?如果装进去第i件商品,那么装入之前是什么状态,肯定是V(i-1,j-w(i))。由于最优性原理(上文讲到),V(i-1,j-w(i))就是前面决策造成的一种状态,后面的决策就要构成最优策略。两种情况进行比较,得出最优。
代码实现
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <iostream>
#include <string>
#include <map>
using namespace std;
#define MAX_N 10000
typedef struct Node
{
int t;
int val;
friend bool operator < (Node a, Node b)
{
if (a.t == b.t)
{
return a.val < b.val;
}
return a.t < b.t;
}
}node;
int main()
{
int T, N, X;
node game[MAX_N];
cin >> T;
for (int ca = 0; ca < T; ca++)
{
int dp[1001][12] = { {0} };
//dp = (int(*)[MAX_N])malloc(N * MAX_N * sizeof(int));
//memset(dp, 0, 1001 * 12 * sizeof(int));
memset(game, 0, MAX_N * sizeof(node));
cin >> N >> X;
for (int i = 0; i < N; i++)
{
cin >> game[i].val >> game[i].t;
}
sort(game, game + N);
for (int i = game[0].t; i <= X; i++)
{
for (int j = 1; j <= N; j++)
{
if (game[j].t <= i)
{
dp[i][j] = max(dp[i - game[j].t][j] + game[j].val, dp[i][j - 1]);
}
}
}
cout << dp[X][N] << endl;
}
return 0;
}