矩阵连乘问题 DP

题目大意：要你求矩阵A1*A2*A3*A4*A5*A6…………*An的最优解法，就是求乘的次序，因为一个i*j的矩阵*一个j*k的矩阵时间复杂度为i*j*k，通过适当的改变运算顺序，可以使原来的问题规模变小。这里给你若干矩阵的行列数，求计算顺序。

解题思路：一道典型的dp，算法老师在课上讲的，然后留的作业，不难。令dp[i][j],为计算从第i到j这段区间的矩阵所需的最小次数，其可以由dp[i+1][j]+相应3个行列数的乘积更新来，同理也可以从右边更新。本题要求的是计算次序，而不是最优解的大小，所以在算出每个区间的最优解的时候，都要把他划分的位置记录下来以便以后查阅，我记录到了goback[][]。之后就是反向推导出计算顺序，我用了个Back（）记录下了相关信息到ans[][]里，最后output，书上的样例能过，但是不知道有没有bug

/*测试数据
6
30 35
35 15
15 5
5 10
10 20
20 25
*/


#include <iostream>
#include <stdio.h>
#include <math.h>
#include <stdlib.h>
#include <string>
#include <string.h>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <stack>
using namespace std;
typedef long long LL;
const int INF=0x7fffffff;
const int MAX_N=1009;

int N,C;
int A[MAX_N][2];
int dp[MAX_N][MAX_N];
int goback[MAX_N][MAX_N];
int ans[MAX_N][3];

int solve(int x,int y){
    if(dp[x][y]!=-1)return dp[x][y];
    if(x==y)return dp[x][y]=0;
    if(x+1==y)return dp[x][y]=A[x][0]*A[x][1]*A[y][1];
    int ans=INF,t;
    for(int i=x;i<=y-1;i++){
        t=ans;
        ans=min(ans,solve(x,i)+solve(i+1,y)+A[x][0]*A[i][1]*A[y][1]);
        if(t!=ans){
            goback[x][y]=i;
        }
    }
    return dp[x][y]=ans;
}

void Back(int x,int y){
    if(y-x==0)return;
    if(y-x==1){
        ans[C][1]=x;
        ans[C][2]=y;
        C++;
        return;
    }
    ans[C][1]=x;
    ans[C][2]=y;
    C++;
    Back(x,goback[x][y]);
    Back(goback[x][y]+1,y);
}

void output(){
    int ct=C-1;
    for(int i=1;i<=C-1;i++){
        printf("第%d步：计算",i);
        printf("%d~%d\n",ans[ct][1],ans[ct][2]);
        ct--;
    }
}

int main(){
    printf("输入矩阵的个数：");
    while(scanf("%d",&N)!=EOF){
        C=1;
        memset(dp,-1,sizeof(dp));
        printf("下面输入矩阵的维数，共%d行\n",N);
        for(int i=1;i<=N;i++){
            scanf("%d%d",&A[i][0],&A[i][1]);
        }
        cout<<"乘积为"<<solve(1,N)<<endl;
        Back(1,N);
        output();


        printf("\n输入矩阵的个数：");
    }
    return 0;
}