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
#include
#include
#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
",ans-1);
    }
}