hdu 5074

思路: 很容易想到dp  我们只需要维护 以第i个数以 j 结尾的最大值就好了,结果就可以直接转移了 相当于每次两个for循环枚举前一个和后一个,我是分四种情况: 

当前是否为-1 ,前一个是否为-1 

#include<bits/stdc++.h>
using namespace std;
int a[1000];
int b[200][200];
int dp[200][200];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        for(int  i =1; i<=m; i++)
            for(int j =1; j<=m; j++)
                scanf("%d",&b[i][j]);
        for(int i = 1; i<=n; i++)
        {
            scanf("%d",&a[i]);
        }
        memset(dp,0,sizeof(dp));
        for(int i= 2; i<=n; i++)
        {
            if(a[i]==-1)
            {
                if(a[i-1]==-1)
                {
                    for(int j = 1; j<=m; j++)
                    {
                        for(int k = 1; k<=m; k++)
                            dp[i][j] = max(dp[i][j],dp[i-1][k]+b[k][j]);
                    }
                }
                else
                {
                    for(int j = 1; j<=m; j++)
                        dp[i][j] = max(dp[i][j],dp[i-1][a[i-1]]+b[a[i-1]][j]);
                }
            }
            else
            {

                if(a[i-1]==-1)
                {
                    for(int k = 1; k<=m; k++)
                        dp[i][a[i]] = max(dp[i-1][k]+b[k][a[i]],dp[i][a[i]]);
                }
                else
                {
                    dp[i][a[i]] = max(dp[i][a[i]],dp[i-1][a[i-1]]+b[a[i-1]][a[i]]) ;
                }
            }
        }
        int ans = 0;
        for(int i = 1; i<=m; i++)
            ans = max(ans, dp[n][i]);
        cout<<ans<<endl;
    }
    return 0;
}


猜你喜欢

转载自blog.csdn.net/theflowerofac/article/details/52781596