D. Yet Another Minimization Problem

题目D. Yet Another Minimization Problem

参考题解：大佬博客

考点：数学公式转换 + 背包
$$\sum _{i=1}^n\sum _{j=i+1}^n(a_i+a_j{)}^2=(a_1+a_2+\cdot \cdot \cdot +a_{n-1}+a_n{)}^2+(n-2)(a_1^2+a_2^2+\cdot \cdot \cdot +a_{n-1}^2+a_n^2)=(n-2)\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)\sum _{i=1}^n{a}_i^2+(\sum _{i=1}^n{a}_i{)}^2+(n-2)\sum _{i=1}^n{b}_i^2+(\sum _{i=1}^n{b}_i{)}^2$$

通过式子可以发现只需考虑 $(a_1+a_2+\cdot \cdot \cdot +a_{n-1}+a_n{)}^2+(b_1+b_2+\cdot \cdot \cdot +b_{n-1}+b_n{)}^2$ 的最小值。

设 $sum={\sum }_{i=1}^n(a_i+b_i)$ ，求出所有可能 $a_i$ 与 $b_i$ 交换后的 $suma={\sum }_{i=1}^n{a}_i$ ， 以及 $sumb=sum-suma$

$ans=min((n-2)({\sum }_{i=1}^n{a}_i^2+{\sum }_{i=1}^n{b}_i^2)+sum{a}^2+sum{b}^2)$

如何快速的计算出 $suma$ 的值？

采用分组背包的思路，$dp[i][j]$ 表示在前 $i$ 个数中是否可以组合出值 $j$ ，最后枚举 $dp[n][i]$ ，如果 $dp[n][i]$ 为真，则表示 $suma$ 的值可以为 $i$ ,再代入公式计算出 $ans$ 的最小值。

#include<stdio.h>
#include<bitset>

using namespace std;
const int N = 1e4 + 10;
int a[N],b[N],dp[110][N];

int main(){
	int t;
	scanf("%d",&t);
	while(t --) {
		int n,sum = 0,mul = 0;
		scanf("%d",&n);
		for(int i=1;i<=n;i++) {
			scanf("%d",&a[i]);
			sum += a[i];
			mul += a[i] * a[i];
		}
		for(int i=1;i<=n;i++) {
			scanf("%d",&b[i]);
			sum += b[i];
			mul += b[i] * b[i];
		}
		for(int i=0;i<=n;i++) {
			for(int j=0;j<=sum;j++) {
				dp[i][j] = 0;
			}
		}
		dp[0][0] = 1;
		for(int i=1;i<=n;i++) {
			for(int j=0;j<=sum;j++) {
				if(j >= a[i] && dp[i-1][j-a[i]]) dp[i][j] = 1;
				if(j >= b[i] && dp[i-1][j-b[i]]) dp[i][j] = 1;
			}
		}
		int res = 0x3f3f3f3f;
		for(int i=0;i<=sum;i++) if(dp[n][i]){
			res = min(res,((n - 2) * mul + i * i + (sum - i) * (sum - i)));
		}
		printf("%d\n",res);
	}
	return 0;
}