三维dp(vivo)

在这里插入图片描述

第一次遇到三维的01背包问题,不过还是很快解决了:
AC的代码如下:

//dp[i][j][k]表示当磁盘大小为j,内存大小为k时,前i个应用程序的最佳组合对应的最大用户数(01背包)
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <algorithm>
using namespace std;

/*
* Welcome to vivo !
*/

int getCountOfApp(const char* input) {
	if (NULL == input) {
		return 0;
	}
	int cnt = 0;
	for (int i = 0; input[i] != 0; ++i) {
		if (input[i] == ',') {
			++cnt;
		}
	}
	return cnt / 2;
}

//id start from 0
int getPositionOfApp(const char* input, const int id) {
	int cntOfComma = id * 2 + 1;
	int i = 0;
	for (; input[i] != 0 && cntOfComma != 0; ++i) {
		if (input[i] == ',') {
			--cntOfComma;
		}
	}
	while (input[--i] != ' '&&input[i] != '#');
	return i + 1;
}

#define APP_MAX 1000
#define DP_MAX 2048
int disks[APP_MAX];
int mems[APP_MAX];
int users[APP_MAX];


int solution(const char* input) {
	const int countOfApp = getCountOfApp(input);
	if (0 == countOfApp) {
		return 0;
	}
	int res = 0;

	int disk = 0;
	int mem = 0;

	sscanf(input, "%d %d", &disk, &mem);

	for (int i = 0; i < countOfApp; ++i) {
		const int pos = getPositionOfApp(input, i);
		sscanf(input + pos, "%d,%d,%d", &disks[i], &mems[i], &users[i]);
	}

	// TODO Write your code here!
	vector<vector<vector<int>>>dp(countOfApp + 1, vector<vector<int>>(disk + 1, vector<int>(mem + 1, 0)));
	for (int i = 1; i <= countOfApp; i++)
		for (int j = 0; j <= disk; j++)
			for (int k = 0; k <= mem; k++)
			{
				if (j >= disks[i - 1] && k >= mems[i - 1])
				{
					dp[i][j][k] = max(dp[i - 1][j][k], dp[i - 1][j - disks[i - 1]][k - mems[i - 1]] + users[i - 1]);
				}
				else
				{
					dp[i][j][k] = dp[i - 1][j][k];
				}
			}
	res = dp[countOfApp][disk][mem];

	return res;
}

int main(int argc, char* args[])
{
	char input[10000];
	cin.getline(input, 10000);
	cout << solution(input) << endl;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值