1. 程式人生 > 實用技巧 >洛谷P1854花店櫥窗佈置-題解

洛谷P1854花店櫥窗佈置-題解

原題:

思路:

考慮DP

設f[i][j]為考慮前i行,第i行選第j個的最大值

則f[i][j]=max(f[i-1][j])+d[i][j]

這道題由於有列舉成分在,如果用DFS就會超時

程式碼:

#include <bits/stdc++.h>
using namespace std;
int n,m;
int pre[150][150];
int a[150][150];
int f[150][150];
void print(int x,int y)
{
    if(pre[x][y]==y)
    {
        cout << y << ' ';
        return;
    }
    print(x-1,pre[x][y]);
    cout << y << ' ';
}
int main()
{
    cin >> n >> m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            cin >> a[i][j];
    for(int i=1;i<=m-n;i++) 
    {
        f[1][i]=a[1][i];
        pre[1][i]=i;
    }
    for(int i=2;i<=n;i++)
        for(int j=i;j<=m-n+i;j++)
            for(int k=1;k<j;k++)
                if(f[i][j]<f[i-1][k]+a[i][j])
                {
                    f[i][j]=f[i-1][k]+a[i][j];
                    pre[i][j]=k;
                }
    int tx=n,ty,ans=0;
    for(int i=n;i<=m;i++)
        if(f[n][i]>ans)
        { 
            ans=f[n][i]; 
            ty=i;
        }
    printf("%d\n",ans);
    print(tx,ty);
    return 0;
}