D. Yet Another Minimization Problem

愚末语

已于 2022-02-17 21:11:58 修改

阅读量681

点赞数 1

分类专栏： cf 文章标签：动态规划算法

于 2022-02-13 20:38:03 首次发布

本文链接：https://blog.csdn.net/weixin_50547586/article/details/122913559

版权

cf 专栏收录该内容

28 篇文章 3 订阅

订阅专栏

本文讨论了一场CF比赛中关于动态规划的问题，作者分享了从暴力枚举到迭代求解最优和的过程，重点讲解了如何通过化简式子和迭代方法找到接近最优的和，同时提到了使用STL库的accumulate函数简化求和操作。

摘要由CSDN通过智能技术生成

传送门：CF

前言：过完年的这么几把，把把掉大分，从1850调到1700，真实酸爽。这场D的dp是一个很妙的点（~~至少我不会~~），看了小t 的代码后，醍醐灌顶（~~小t如药也，善读可以医愚~~）。

正文：

首先看下数据范围，给了2秒的实现加上n为100，我初解的时候觉得是一个暴力枚举的题，所以压根就没往化简式子+dp上面想。所以掉大分。

通过化简式子，可以得到，最优的时候是 $\sum a=\frac{(\sum a+\sum b)}{2}$

当然，很难得到最完美的情况，所以只要尽量接近就行。

然而我们该怎么找到这个最优的和呢？

我最初的想法是暴力，但是MLE了（~~我是傻逼~~）

正解是通过迭代，AC代码如下：

#include<bits/stdc++.h>
using namespace std;
#define ll long long

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int T; 
    cin>>T;
	while(T--)
	{
        int n;
        cin>>n;
        vector<int>a(n),b(n);
        for(int i=0;i<n;i++)cin>>a[i];
        for(int i=0;i<n;i++)cin>>b[i];
        if(n==1){
            cout<<0<<'\n';
            continue;
        }
        ll sum = accumulate(a.begin(),a.end(),0)+accumulate(b.begin(),b.end(),0);
        vector<ll>dp(sum+1);
        dp[0]=1;
        for(int i=0 ;i<n;i++){
            for(int j = sum ; j >= 0;j--){
                if(dp[j]){
                    dp[j+a[i]]=1;
                    dp[j+b[i]]=1;
                    dp[j]=0;
                }
            }
        }
        
        ll kase = 1e18;
        ll sum1=0;
        for(int i=0;i<=sum;i++){
            if(dp[i]==0)continue;
            if(abs(sum-i*2)<kase){
                sum1=i;
                kase = abs(sum-i*2);
            }
        }
        ll ans=0;
      //  cout<<sum1<<'\n';
        ans+=sum1*sum1+(sum-sum1)*(sum-sum1);
        for(int i =0 ;i<n;i++){
            ans+=(n-2)*a[i]*a[i];
            ans+=(n-2)*b[i]*b[i];
        }
        cout<<ans<<'\n';
        
	}
	return 0;
}

这里又学了一个STL，

可以用accumulate（）来直接求解一个数组的和。