P1401 矩阵连乘问题

P1401 矩阵连乘问题

注意k的分段枚举。（区间dp真是太好玩啦

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e3+6;
struct sq{
    int r,c;
}s[maxn];
ll  a[maxn],dp[maxn][maxn];
int main(int argc, char const *argv[]) {
    int n;
    while (scanf("%d",&n)!=EOF) {
        memset(dp,0,sizeof dp);
        int flag=0;
        for(int i=0;i<n;i++){
            scanf("%d %d",&s[i].r,&s[i].c);
        }
        a[0]=s[0].r;
        for(int i=1;i<n;i++){
            if(s[i].r!=s[i-1].c) {flag=1;break;}
            a[i]=s[i].r;
        }
        a[n]=s[n-1].c;
        if(flag) {std::cout << "invalid argument" << '\n';continue;}
        for(int cnt=2;cnt<=n;cnt++){
            for(int i=0;i+cnt<=n;i++){
                int j=i+cnt;
                dp[i][j]=dp[i][j-1]+a[i]*a[j-1]*a[j];
                for(int k=i+1;k<j;k++){
                    dp[i][j]=min(a[i]*a[k]*a[j]+dp[i][k]+dp[k][j],
                        dp[i][j]);
                }
            }
        }
        std::cout <<dp[0][n] << '\n';
    }
    return 0;
}