atcoder 163E——DP

题目链接

题意：n个小孩，每个人有一个位置pi，还有一个权值Ai。求任意交换位置之后，每个小孩权值✖交换位置距离之和最大是多少。

解析：贪心的思路，权值越大的小孩就越往边上去。仔细想一下应该是一个区间DP

$dp[l][r]=max(dp[l][r],max(dp[l+1][r]+A_i\times |p_i-l|，dp[l][r-1]+A_i\times |p_i-r|))$

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define int long long
int dp[2010][2010];
int a[2010];
struct node
{
    int first;
    int second;
}c[5001];


bool cmp(node p1,node p2)
{
    return p1.second>p2.second;
}

int work(int now,int l,int r)
{
	if (l>r)  return c[now].second;
	if (dp[l][r]!= -1)  return dp[l][r];

	int f=work(now+1,l+1,r)+c[now].second*abs(c[now].first-l);
	int s=work(now+1,l,r-1)+c[now].second*abs(c[now].first-r);
	dp[l][r]=max(dp[l][r], max(f,s) );

	return dp[l][r];
}
int main()
{
    int n;
    cin>>n;
    memset(dp,-1,sizeof(dp));
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        c[i].second = a[i];
        c[i].first = i;
    }
    sort(c+1,c+1+n,cmp);
    cout<<work(1,1,n);
   
}

#define int long long 是真的骚啊