牛客137最后贪心小结论

题目

https://ac.nowcoder.com/acm/contest/130843/F

code

#include<bits/stdc++.h>
using namespace std;
//"O campeão tem nome, e se chama Charles Oliveira!"
#define int long long
#define endl '\n'
#define ep emplace
#define pob 
#define ll long long
#define pb push_back
#define pof pop_front
#define pob pop_back
#define all(a) a.begin(),a.end()
#define rall(a) a.rbegin(),a.rend()
#define mod 998244353
#define MOD 1000000007
const int N=200005;
const int INF=1e18;
using ld = long double;
using ui = unsigned;
using ull = unsigned long long;
using i128 = __int128;

/*
搬运问题贪心结论：当条件类似于把一个数组变成另外一个数组，+1 -1，且代价是数组中两个元素之间的下标的差值的
最小代价就是两个数组前缀和之差
类似于prevA[i]-prevB[i];
*/

void solve(){
    int n;cin>>n;
    vector<int>a(n+1);
    for(int i=1;i<=n;i++){
        cin>>a[i];
        a[i]=a[i]+a[i-1];
    }
    vector<vector<int>>dp(n+1,vector<int>(a[n]+1,INF));//含义是，处理到前i个数字的时候，总和是j的最小代价
    dp[0][0]=0;
  for(int p=0;p*p<=a[n];p++){//枚举所有可用的完全平方数字
        int x=p*p;
    for(int i=1;i<=n;i++){
        for(int j=x;j<=a[n];j++){
            dp[i][j]=min(dp[i][j],dp[i-1][j-x]+abs(j-a[i]));
        }
    }
  }
  if(dp[n][a[n]]==INF)dp[n][a[n]]=-1;
  cout<<dp[n][a[n]]<<endl;
}

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int t=1;
    cin>>t;
    while(t--)solve();
}

分析

小结论

搬运问题贪心结论：当条件类似于把一个数组变成另外一个数组，+1 -1，且代价是数组中两个元素之间的下标的差值的
最小代价就是两个数组前缀和之差
类似于prevA[i]-prevB[i];
###分析