Problem F. Fighting Against Monsters dp

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+100;
const ll inf=(1ll<<61)-1;
ll dp[105][205][205],atk[4],hp[4],newatk[3],cost;

ll gao(ll H) {
    ll t=(sqrtl(1+8*H)-1)/2;
    while(t*(t+1)/2>H)t--;
    while(t*(t+1)/2<H)t++;
    return t;
}

int main() {
    int cas;cin>>cas;
    while(cas--) {
        scanf("%lld%lld%lld%lld%lld%lld",&hp[1],&hp[2],&hp[3],&atk[1],&atk[2],&atk[3]);
        for(int i=0;i<=100;i++)
            for(int j=0;j<200;j++)
                for(int k=0;k<200;k++)
                    dp[i][j][k]=inf;
        dp[0][0][0]=0;
        ll ans=inf;
        for(int i=1;i<=100;i++)
            for(int j=0;j<200;j++)
                for(int k=0;k<200;k++) if(dp[i-1][j][k]!=inf) {
                    cost=atk[1]*(j<hp[1])+atk[2]*(k<hp[2])+atk[3]*(i*(i-1)/2-k-j<hp[3]);//先算受到的伤害
                    if(j<hp[1]) dp[i][j+i][k]=min(dp[i][j+i][k],dp[i-1][j][k]+cost);
                    if(k<hp[2]) dp[i][j][k+i]=min(dp[i][j][k+i],dp[i-1][j][k]+cost);
                    if(i*(i-1)/2-k-j<hp[3]) dp[i][j][k]=min(dp[i][j][k],dp[i-1][j][k]+cost);
        }

        for(int i=1;i<=100;i++)
            for(int j=0;j<200;j++)
                for(int k=0;k<200;k++) if(dp[i][j][k]!=inf&&j>=hp[1]&&k>=hp[2]&&i*(i+1)/2-k-j>=hp[3])
                    ans=min(ans,dp[i][j][k]);

        for(int i=100;i<=100;i++)
            for(int j=0;j<200;j++)
                for(int k=0;k<200;k++) if(dp[i][j][k]!=inf) {
                    newatk[1]=atk[1]*(j<hp[1]);
                    newatk[2]=atk[2]*(k<hp[2]);
                    if(newatk[1]>newatk[2])swap(newatk[1],newatk[2]);
                    ll t=gao(max(1ll*i*(i+1)/2,j+k+hp[3]));
                    ans=min(ans,dp[i][j][k]+ (t-i)*atk[3]+(t-i+1)*newatk[2]+(t-i+2)*newatk[1]);
        }
        printf("%lld\n",ans);
    }
    return 0;
}