Gym - 102056I——I - Misunderstood … Missing （DP）

題目鏈接：https://codeforces.com/gym/102056/problem/I

思路：倒過來從後往前dp，dp【i】【j】【k】表示從最後一個開始到第i個選擇j次a，所有a到當前i的距離和爲k的最大值

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define ll long long
using namespace std;
ll t,n;
ll dp[2][105][5500];
struct node
{
    ll a,b,c;
} A[105];
int main()
{
    for(scanf("%lld",&t); t; t--)
    {
        memset(dp,0,sizeof(dp));
        scanf("%lld",&n);
        for(ll i=n; i>0; i--)
            scanf("%lld%lld%lld",&A[i].a,&A[i].b,&A[i].c);
        ll d=1;
        dp[0][0][0]=1;
        for(ll i=1; i<=n; i++,d^=1)
        {
            for(ll j=0; j<=i; j++)
            {
                for(ll k=0; k<=i*j&&k<=5300; k++)
                {
                    if(j>0&&k-j+1>=0&&dp[d^1][j-1][k-j+1]) dp[d][j][k]=max(dp[d][j][k],dp[d^1][j-1][k-j+1]+A[i].a);
                    if(k-j>=0&&dp[d^1][j][k-j])dp[d][j][k]=max(dp[d][j][k],dp[d^1][j][k-j]+A[i].b*k);
                    if(k-j>=0&&dp[d^1][j][k-j])dp[d][j][k]=max(dp[d][j][k],dp[d^1][j][k-j]+A[i].c*j);
                }
            }
        }
        ll ans=0;
        for(ll i=0; i<=n; i++)
            for(ll j=0; j<=5300; j++)
                ans=max(ans,dp[d^1][i][j]);
        printf("%lld\n",ans-1);
    }
}