动态规划实验

题意：

根据一张有向图，寻找最短路径，用动态规划做

思路：

将所有点都存储起来，用数塔类似的方法求解即可

代码1--由后向前

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
const int maxn=1000;
const int inf=0x3f3f3f3f;
struct Node
{
    int num;
    int x[100];
} node[maxn];

int e[maxn][maxn];
int e_num;
int k;

int DP(int p,int h)
{
    if(p==0)
    {
        return 0;
    }
    int minn=inf;

        int ii=node[p].x[h];
        for(int j=0; j<node[p-1].num; j++)
        {
            int jj=node[p-1].x[j];
            if(e[jj][ii]!=-1)
            {
                printf("  %d  %d\n",jj,ii);
                minn=min(minn,DP(p-1,j)+e[jj][ii]);
            }
        }
    return minn;
}
int main()
{
    int a,b,c;

    printf("输入k的值\n");
    scanf("%d",&k);
    for(int i=0; i<k; i++)
    {
        printf("第%d节点有几个节点\n",i);
        scanf("%d",&node[i].num);
        printf("输入这%d个节点\n",node[i].num);
        for(int j=0; j<node[i].num; j++)
        {
            scanf("%d",&node[i].x[j]);
        }
    }

    memset(e,-1,sizeof(e));

    printf("输入图中的边的个数\n");
    scanf("%d",&e_num);
    printf("输入这些边的起点和中点\n");
    for(int i=0; i<e_num; i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        e[a][b]=c;
    }
   // printf("%d\n",node[k-1].x[0]);
    printf("%d",DP(k-1,0));

    return 0;
}

代码2--由前向后

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
const int maxn=1000;
const int inf=0x3f3f3f3f;
struct Node
{
    int num;
    int x[100];
} node[maxn];

int e[maxn][maxn];
int dp[maxn][maxn];
int e_num;
int k;

int main()
{
    int a,b,c;

    printf("输入k的值\n");
    scanf("%d",&k);
    for(int i=0; i<k; i++)
    {
        printf("第%d节点有几个节点\n",i);
        scanf("%d",&node[i].num);
        printf("输入这%d个节点\n",node[i].num);
        for(int j=0; j<node[i].num; j++)
        {
            scanf("%d",&node[i].x[j]);
        }
    }

    memset(e,-1,sizeof(e));

    printf("输入图中的边的个数\n");
    scanf("%d",&e_num);
    printf("输入这些边的起点和中点\n");
    for(int i=0; i<e_num; i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        e[a][b]=c;
    }
    //printf("%d\n",node[k-1].x[0]);
    for(int i=0;i<node[0].num;i++)
    {
        dp[0][i]=0;
    }
    for(int i=1;i<k;i++)
    {
        for(int j=0;j<node[i].num;j++)
        {
            int jj=node[i].x[j];
            dp[i][j]=inf;
            for(int k=0;k<node[i-1].num;k++)
            {
                int kk=node[i-1].x[k];
                if(e[kk][jj]!=-1)
                {
                    dp[i][j]=min(dp[i-1][k]+e[kk][jj],dp[i][j]);
                }
            }
        }
    }
    printf("%d\n",dp[k-1][0]);


    return 0;
}

测试样例

5
1
1
4
2 3 4 5
3
6 7 8
3
9 10 11
1
12
21
1 2 9
1 3 7
1 4 3
1 5 2
2 6 4
2 7 2
2 8 1
3 6 2
3 7 7
4 8 11
5 7 11
5 8 8
6 9 6
6 10 5
7 9 4
7 10 3
8 10 5
8 11 6
9 12 4
10 12 2
11 12 5

相关链接