蓝桥杯 算法提高 矩阵乘法

题目如下：

问题描述

　　有n个矩阵，大小分别为a0*a1, a1*a2, a2*a3, ..., a[n-1]*a[n]，现要将它们依次相乘，只能使用结合率，求最少需要多少次运算。
　　两个大小分别为p*q和q*r的矩阵相乘时的运算次数计为p*q*r。

输入格式

　　输入的第一行包含一个整数n，表示矩阵的个数。
　　第二行包含n+1个数，表示给定的矩阵。

输出格式

　　输出一个整数，表示最少的运算次数。

样例输入

3
1 10 5 20

样例输出

150

数据规模和约定

　　1<=n<=1000, 1<=ai<=10000。

-----分割线-----

　　dp,类似填表的做法，代码如下：

#include<stdio.h>
#include<stdlib.h>
long long int a[1001];
long long int matrix_DP(int n)
{
    long long int t,min;
    int l,i,j,k;
    long long int *b=malloc(sizeof(long long int)*1001*1001);
    for(i=0;i<=1000;i++)
        for(j=0;j<=1000;j++)
            *(b+i*1000+j)=0;
    for(l=2;l<=n;l++)
    {
        for(i=1,j=l;j<=n;i++,j++)
        {
            min=1e18;
            for(k=i;k<j;k++)
            {
                t=*(b+i*1000+k)+*(b+(k+1)*1000+j)+a[i-1]*a[k]*a[j];
                if(t<min)
                    min=t;
            }
            *(b+i*1000+j)=min;
        }
    }
    min=*(b+1000+n);
    free(b);
    return min;
}
int main()
{
    int n,i;
    scanf("%d",&n);
    for(i=0;i<n+1;i++)
        scanf("%I64d",a+i);
    printf("%I64d",matrix_DP(n));
    return 0;
}