Yet Another Minimization Problem（数论/dp/01背包）

题目
题意：给定两个长度为n的数组$a,b$，可以交换对应位置的$a_i,b_i,1<=i<=n$。问能得到的、最小的 ${\sum }_{i=1}^n{\sum }_{j=i+1}^n(a_i+a_j{)}^2$+${\sum }_{i=1}^n{\sum }_{j=i+1}^n(b_i+b_j{)}^2$是多少。

参考

思路：令$s={\sum }_{i=1}^n{a}_i$，则${\sum }_{i=1}^n{\sum }_{j=i+1}^n(a_i+a_j{)}^2=(n-1)\ast {\sum }_{i=1}^n{a_i}^2+{\sum }_{i=1}^n{a}_i\ast (s-a_i)=(n-2)\ast {\sum }_{i=1}^n{a_i}^2+{\sum }_{i=1}^n{a_i}^2$

所以${\sum }_{i=1}^n{\sum }_{j=i+1}^n(a_i+a_j{)}^2$+${\sum }_{i=1}^n{\sum }_{j=i+1}^n(b_i+b_j{)}^2=(n-2)\ast {\sum }_{i=1}^n({a_i}^2+{b_i}^2)+({\sum }_{i=1}^n{a}_i{)}^2+({\sum }_{i=1}^n{b}_i{)}^2$

由于$(n-2)\ast {\sum }_{i=1}^n({a_i}^2+{b_i}^2)$是固定的，原题等价于最小化$({\sum }_{i=1}^n{a}_i{)}^2+({\sum }_{i=1}^n{b}_i{)}^2$

令$suma={\sum }_{i=1}^n{a}_i,sumb={\sum }_{i=1}^n{b}_i$，因为$sum=suma+sumb$是固定的，要使$sum{a}^2+sum{b}^2$尽可能小，要让$suma,sumb$尽可能平均（周长相同的情况下，正方形是面积最小的四边形。代数上也很好证明）。

令$c_i=max(a_i,b_i)-min(a_i,b_i)$，$s=\sum c_i\ast (1取/0不取)$，则$suma=s+{\sum }_{i=1}^{n}min(a_i,b_i),sumb=sum-suma$。
要使suma,sumb接近，s要尽量接近总量$\sum c_i$的一半，则相当于从背包$m=(\sum c_i)/2$中尽可能装满物品。01背包模型。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 200010;

int t;
int a[maxn], b[maxn], n;
int dp[maxn];
void solve() {
	scanf("%d", &n);
	int sum = 0, res = 0;
	int all = 0, ans = 0;
	for (int i = 0; i < n; ++i) {
		scanf("%d", &a[i]);
		all += a[i];
		ans += a[i] * a[i];
	}
	for (int i = 0; i < n; ++i) {
		scanf("%d", &b[i]);
		all += b[i];
		ans += b[i] * b[i];
		if (a[i] > b[i]) {
			swap(a[i], b[i]);
		}
		res += a[i];
		b[i] -= a[i];
		sum += b[i];
	}
	
	// 01背包 
	memset(dp, 0, sizeof(dp));
	int m = sum / 2;
	for (int i = 0; i < n; ++i) {
		for (int j = m; j >= b[i]; --j) {
			dp[j] = max(dp[j], dp[j-b[i]] + b[i]);
		}
	}
	
	int suma = res + dp[m];
	int sumb = all - suma;
	ans = ans * (n - 2) + suma * suma + sumb * sumb;
	printf("%d\n", ans);
}
int main() {
	scanf("%d", &t);
	while (t--) {
		solve();
	}
} 

另一种思路：定义$d{p}_{iw}$，表示取前i个数，是否能可达$w$这个分，可达则为$true$，否则为$false$。
$dp[i][w]=max(dp[i-1][w-a[i]],dp[i-1][w-b[i]])$
最后，从$d{p}_{n}w=true$的所有$w$中，suma的值，取$|suma-sumb|$最小的那个。计算对应的答案值。