1113. Integer Set Partition 解析

最新推荐文章于 2022-11-15 17:07:48 发布

原创最新推荐文章于 2022-11-15 17:07:48 发布 · 665 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#PAT

PAT 专栏收录该内容

138 篇文章

订阅专栏

本文介绍了一种通过排序和从两端向中间累加元素值的方法来平衡划分数组的算法。该算法适用于需要将数组划分为两个子集的问题，使得这两个子集在元素数量和总和上尽可能接近。

排序然后两头开始加，双数一边一半就好，单数看给左边还是右边。

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int n;
vector <int> list;


int main() {
	cin >> n;

	int temp;
	for (int i = 0; i < n; i++) {
		scanf("%d",&temp);
		list.push_back(temp);
	}

	sort(list.begin(), list.end());

#ifdef _DEBUG
	for (int i = 0; i < list.size(); i++) {
		cout << list[i] << endl;
	}
#endif

	//处理分组
	int s1 = 0, s2 = 0;
	int n1 = 0, n2 = 0;
	int p1 = 0, p2 = list.size() - 1;
	while (p1 < p2) {
		s1 += list[p1];
		s2 += list[p2];
		n1++, n2++, p1++, p2--;
	}

	if (p1 == p2) {//奇数情况
		int ts1 = s1 + list[p1], ts2 = s2 + list[p2];
		if (abs(ts1 - s2) > abs(ts2 - s1)) {
			s1 = ts1, n1++;
		}
		else {
			s2 = ts2, n2++;
		}
	}

	cout << abs(n1 - n2) << " " << abs(s1 - s2) << endl;

	return 0;

}