总公司拥有M台 相同 的高效设备,准备分给下属的N个分公司。
各分公司若获得这些设备,可以为国家提供一定的盈利。盈利与分配的设备数量有关。
问:如何分配这M台设备才能使国家得到的盈利最大?
求出最大盈利值。
分配原则:每个公司有权获得任意数目的设备,但总台数不超过设备数M。
输入格式
第一行有两个数,第一个数是分公司数N,第二个数是设备台数M;
接下来是一个N*M的矩阵,矩阵中的第 i 行第 j 列的整数表示第 i 个公司分配 j 台机器时的盈利。
输出格式
第一行输出最大盈利值;
接下N行,每行有2个数,即分公司编号和该分公司获得设备台数。
答案不唯一,输入任意合法方案即可。
数据范围
1≤N≤10,
1≤M≤15
输入样例:
3 3
30 40 50
20 30 50
20 25 30
输出样例:
70
1 1
2 1
3 1
历程:由于是dp专题,很自然的的用dp做,很快就写完了,但是我发现这道题还有一问,求具体方案,emm,还没遇到过呢,算了,先去研究下前边的求背包问题具体方案板子题吧,很快我把思路模拟出来啦,于是在配合板子写,A掉之后回来也很顺利的AC掉机器分配这一道,但是后来听说这道题还可以用dfs来做,看了下,应该是可以,用dfs做了下,用时700+, 但是我copy的评论区的代码只用时100+,心里不平衡,发现dfs优化真的很重要,牵一发而动全身!!
思路:首先先介绍下分组背包解法f[i][j]
表示在前i组中选,数量不超过j
的最大价值,将所有的初始值设置为0
,我们的状态转移方程为
f[i][j] = max(f[i - 1][j - v[i, 1]] + w[i, 1],f[i - 1][j - v[i,2]] + w[i, 2]..., f[i][j])
其中v[i, j]
代表第i
组的第j
件物品,w[i, j]
同理。
代码也很简洁。
#include <algorithm>
#include <iostream>
using namespace std;
const int N = 20;
int map[N][N], f[N][N];
int main()
{
int n, m; cin >> n >> m;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++) scanf("%d", &map[i][j]);
for (int i = n; i >= 1; i--)
{
for (int j = m; j >= 1; j--)
{
for (int k = 0; k <= j; k++)
{
f[i][j] = max(f[i][j], f[i + 1][j - k] + map[i][k]);
}
}
}
cout << f[1][m] << endl;
int k = m;
for (int i = 1; i <= n; i++)
{
for (int j = 0; j <= m; j++)
{
if (f[i][k] == f[i + 1][k - j] + map[i][j])
{
printf("%d %d\n", i, j);
k -= j;
break;
}
}
}
return 0;
}
我开始写的dfs代码, 用时700+ ms
#include <iostream>
using namespace std;
const int N = 20;
int map[N][N], n, m, ans, idx, cnt[N], cnt_ans[N];
int dfs(int res, int de, int idx)//价值,个数,层数
{
if(de == m)
{
if(ans < res)
{
ans = res;
for(int i = 1; i <= n; i ++) cnt_ans[i] = cnt[i];
}
ans = max(ans, res);
}
if(idx > n) return 0;
for(int i = 0; i <= m ; i ++)
{
if(de + i <= m)
{
cnt[idx] = i;
dfs(res + map[idx][i], de + i, idx + 1);
cnt[idx] = 0;
}
}
return 0;
}
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i ++)
for(int j = 1; j <= m; j ++) scanf("%d", &map[i][j]);
dfs(0, 0, 1);
cout << ans << endl;
for(int i = 1; i <= n ; i ++)
{
cout << i << ' ' << cnt_ans[i] << endl;
}
return 0;
}
后来看的讨论区的代码, 进行了修改,很多值得我学习的地方~
#include <iostream>
using namespace std;
const int N = 20;
int map[N][N], n, m, ans, idx, cnt[N], cnt_ans[N];
void dfs(int res, int de, int idx)//价值,个数,层数
{
if(idx == n)
{
cnt[n] = m - de;
res += map[n][m - de];
if(ans < res)
{
ans = res;
for(int i = 1; i <= n; i ++) cnt_ans[i] = cnt[i];
}
return ;
}
for(int i = 0; i <= m - de; i ++)
{
cnt[idx] = i;
dfs(res + map[idx][i], de + i, idx + 1);
}
return ;
}
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i ++)
for(int j = 1; j <= m; j ++) scanf("%d", &map[i][j]);
dfs(0, 0, 1);
printf("%d\n", ans);
for(int i = 1; i <= n ; i ++)
{
printf("%d %d \n",i, cnt_ans[i]);
}
return 0;
}
说一下我的思考,都知道dfs搜索其实是搜索的一棵树,这棵树的大小是呈指数级别的,所以比其减少树的枝干,不如减少树的深度,嗯,牢记~