P2066 机器分配 题解

· · 题解

#include<bits/stdc++.h>
#define int long long
using namespace std;
int a[1005][1005],dp[1005][1005];
int ans[100005];
signed main()
{
    int n,m;
    cin>>n>>m;
    for(int i=n;i>=1;i--)
    {
        for(int j=1;j<=m;j++)
        {
            cin>>a[i][j];
        }
    }   
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<=m;j++)
        {
            for(int k=0;k<=j;k++)
            {
                dp[i][j]=max(dp[i][j],dp[i-1][j-k]+a[i][k]);
            }
        }
    }
    cout<<dp[n][m]<<'\n';
    int s=m,sum=dp[n][m];
    for(int i=n;i>=1;i--)
    {
        for(int j=0;j<=m;j++)
        {
            if(dp[i-1][s-j]+a[i][j]==sum)
            {
                sum-=a[i][j];
                ans[i]=j;
                s-=j;
                break;
            }
        }
    }
    for(int i=n;i>=1;i--)
        cout<<n-i+1<<" "<<ans[i]<<'\n';
    return 0;
}