面试OR笔试42——划分数组

1 题目及要求

1.1 题目描述

将正整数数组中的数字划分为两组，使得两个子数组的和的差值最小。

很容易知道，如果整个数组的和为sum，则选取的某个子数组的和越接近sum/2，则这两个子数组的和的差值最小越。即从数组中选取某些数字使得其和最接近sum/2时，则这些数字组成的子数组和其他的数字组成的子数组即为所求。所以该命题本质上是一个01背包命题，原命题等价于从n各物品中选取若干个，其体积和不超过sum/2，且价值达到最大（价值和体积一致）。

2 解答

2.1 代码

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

int split(const vector<int>& num, vector<int>& o1 = vector<int>(), vector<int>& o2 = vector<int>()) {
	int n(num.size());
	if (n < 1) return 0;
	int sum(0);
	for (int k1(0);k1 < n;++k1) {			// 计算和
		sum += num[k1];
	}
	int dtc((sum >> 1) + 1);				// 化为背包问题
	vector<vector<int> > dt(n + 1, vector<int>(dtc, 0));
	for (int k1(0);k1 < n; ++k1) {
		for (int k2(0);k2 < dtc; ++k2) {
			k2 < num[k1] ? dt[k1 + 1][k2] = dt[k1][k2] : dt[k1 + 1][k2] = max(dt[k1][k2], dt[k1][k2 - num[k1]] + num[k1]);
		}
	}
	o1.clear(); o2.clear();
	for (int k1(n), k2(dtc-1);0 < k1;--k1) {	// 输出结果
		if (dt[k1][k2] == dt[k1 - 1][k2]) {
			o2.push_back(num[k1 - 1]);
		}
		else {
			o1.push_back(num[k1 - 1]);
			k2 -= num[k1-1];
		}
	}
	return sum - (dt[n][dtc - 1] << 1);			// 返回差值
}


int main() {
	vector<int> v{ 5,4,2,1,3 };
	vector<int> o1, o2;
	cout << "差值 : " << split(v, o1, o2) << endl;
	cout << "分组1: ";
	for (auto i : o1) cout << i << ' ';
	cout << endl;
	cout << "分组2: ";
	for (auto i : o2) cout << i << ' ';
	cout << endl;

	return 0;
}