URAL1029. Ministry（DP+路径）

链接

路径麻烦啊 很多细节 倒回去搜一遍

卡了一节数据库。。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define LL long long
int u,o,path[50010],flag;
LL sum[110][510],dp[110][510],f[110][510];
int m,n;
void dfs(int u,int v)
{
    if(flag)
    return ;
    int i,j,tt=o;
    if(v==1)
    {
        o++;
        path[o] = u;
        for(i = o ; i > 1 ; i--)
        printf("%d ",path[i]);
        printf("%d
",path[1]);
        flag = 1;
        return ;
    }
    if(dp[v][u]==dp[v-1][u]+f[v][u])
    {
        o = tt;
        path[++o] = u;
        dfs(u,v-1);
    }
    for(i = u-1; i >= 1 ; i--)
    {
        if(dp[v][i]!=dp[v-1][i]+f[v][i])
        continue;
        o = tt;
        if(dp[v][u]==dp[v][i]+sum[v][u]-sum[v][i])
        {
            for(j = u ; j>= i ; j--)
            path[++o] = j;
            dfs(i,v-1);
        }
    }
    for(i = u+1 ; i <= m ; i++)
    {
        if(dp[v][i]!=dp[v-1][i]+f[v][i])
        continue;
        o = tt;
        if(dp[v][u]==dp[v][i]+sum[v][i-1]-sum[v][u-1])
        {
            for(j = u; j <= i; j++)
            path[++o] = j;
            dfs(i,v-1);
        }
    }
    o = tt;
}
int main()
{
    int i,j,g;
    flag=0;
    scanf("%d%d",&n,&m);
    for(i = 1; i <= n ;i++)
    {
        for(j = 1; j <=m ; j++)
        {
            scanf("%lld",&f[i][j]);
            sum[i][j] = sum[i][j-1]+f[i][j];
        }
    }
    for(i = 1 ;i <= n ;i++)
    {
        for(j = 1; j <=m ; j++)
        dp[i][j] = dp[i-1][j]+f[i][j];
        for(j = 1; j <= m ; j++)
        {
            for(g = 1; g < j ;g++)
            dp[i][j] = min(dp[i][j],dp[i][g]+sum[i][j]-sum[i][g]);
            for(g = j+1 ; g <=m ; g++)
            dp[i][j] = min(dp[i][j],dp[i][g]+sum[i][g-1]-sum[i][j-1]);
        }
    }
    LL ans = dp[n][1],u = 1;
    for(i = 1; i <= m ; i++)
    {
        if(ans>=dp[n][i])
        {
            ans = dp[n][i];
            if(dp[n][i]==dp[n-1][i]+f[n][i])
            u = i;
        }
    }
    o++;
    path[o] = u;
    if(n==1)
    printf("%d
",u);
    else
    dfs(u,n-1);
    return 0;
}