3. AI作曲.py

HIM · 2025 年7 月 11 日 12:06

啊啊啊，真没招了，有AC代码（题解）对照着也瞪不出问题

MY CODE

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll dp[105][55], ms[55][55], a[105];
int main(){
    ll t;
    cin >> t;
    while(t--){
        ll n, m;
        cin >> n >> m;
        for(ll i=0;i<m;i++){
            for(ll j=0;j<m;j++){
                cin >> ms[i][j];
            }
        }
        for(ll i=0;i<n;i++){
            cin >> a[i];
        }
        memset(dp, 0, sizeof(dp));
        for(ll i=1;i<n;i++){
            if(a[i]>0&&a[i-1]>0){
                    dp[i][a[i]]=dp[i-1][a[i-1]]+ms[a[i-1]][a[i]];
            }
            else if(a[i]>0&&a[i-1]<0){
                for(ll k=0;k<m;k++)
                    dp[i][a[i]]=max(dp[i][a[i]],dp[i-1][k]+ms[k][a[i]]);
                
            }
            else if(a[i]<0&&a[i-1]>0){
               for(ll j=0;j<m;j++)
                    dp[i][j]=max(dp[i][j],dp[i-1][a[i-1]]+ms[a[i-1]][j]);
            }
            else{
               for(ll j=0;j<m;j++)
                    for(ll k=0;k<m;k++)
                        dp[i][j]=max(dp[i][j],dp[i-1][k]+ms[k][j]);
            }
        }
        ll ans=0;
        for(ll i=0;i<m;i++){
            ans=max(ans, dp[n-1][i]);
        }
        cout << ans << endl;
    }
}

AC CODE

HIM · 2025 年7 月 11 日 12:06

@冯俊骁

HIM · 2025 年7 月 11 日 12:08

@李予劼

CZF2919 · 2025 年7 月 11 日 23:39

数组开小了，而且是else if不是else，可能等于0时是不用操作的

HIM · 2025 年7 月 12 日 00:01

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll dp[105][105], ms[105][105], a[105];
int main(){
    ll t;
    cin >> t;
    while(t--){
        ll n, m;
        cin >> n >> m;
        for(ll i=0;i<m;i++){
            for(ll j=0;j<m;j++){
                cin >> ms[i][j];
            }
        }
        for(ll i=0;i<n;i++){
            cin >> a[i];
        }
        memset(dp, 0, sizeof(dp));
        for(ll i=1;i<n;i++){
            if(a[i]>0&&a[i-1]>0){
                    dp[i][a[i]]=dp[i-1][a[i-1]]+ms[a[i-1]][a[i]];
            }
            else if(a[i]>0&&a[i-1]<0){
                for(ll k=0;k<m;k++)
                    dp[i][a[i]]=max(dp[i][a[i]],dp[i-1][k]+ms[k][a[i]]);
                
            }
            else if(a[i]<0&&a[i-1]<0){
               for(ll j=0;j<m;j++)
                    for(ll k=0;k<m;k++)
                        dp[i][j]=max(dp[i][j],dp[i-1][k]+ms[k][j]);
            }
            else if(a[i]<0&&a[i-1]>0){
                 for(ll j=0;j<m;j++)
                    dp[i][j]=max(dp[i][j],dp[i-1][a[i-1]]+ms[a[i-1]][j]);
            }
        }
        ll ans=0;
        for(ll i=0;i<m;i++){
            ans=max(ans, dp[n-1][i]);
        }
        cout << ans << endl;
    }
}