CodeForces 82 D.Two out of Three（dp）

Description

有一个长度为$n$的顾客队列，服务第$i$个顾客需要$a_i$时间，每次可以从队首的三名顾客中选两位一起服务，所用时间是服务这两名顾客所需时间的较大值，问如何安排可以使得总耗时最小

Input

第一行输入一整数$n$表示顾客数量，之后输入$n$个整数$a_i$表示服务第$i$名顾客的用时

$(1\le n\le 1000,1\le a_i\le 10^6)$

Output

输出最少用时以及按顺序输出每次服务的顾客编号

Sample Input

4
1 2 3 4

Sample Output

6
1 2
3 4

Solution

假设队首两个顾客的编号为$i,j$，那么一个显然的结论是后面队伍顾客的编号从$j+1$开始且连续，如此可以定义$dp[i][j]$表示前两名顾客编号为$i,j$时服务完剩余顾客所需的最少用时，队首三名顾客编号为$i,j,j+1$，根据所选两名顾客的编号不同进行转移即可，通过判断一个状态是由哪个后继状态转移过来即可得到具体方案

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1005;
int n,a[maxn],dp[maxn][maxn];
int dfs(int i,int j)
{
    if(dp[i][j])return dp[i][j];
    if(j==n+1)return dp[i][j]=a[i];
    if(j==n)return dp[i][j]=max(a[i],a[j]);
    dp[i][j]=max(a[i],a[j])+dfs(j+1,j+2);
    dp[i][j]=min(dp[i][j],max(a[i],a[j+1])+dfs(j,j+2));
    dp[i][j]=min(dp[i][j],max(a[j],a[j+1])+dfs(i,j+2));
    return dp[i][j];
}
void output(int i,int j)
{
    if(j==n+1)printf("%d\n",i);
    else if(j==n)printf("%d %d\n",i,j);
    else if(dp[i][j]==max(a[i],a[j])+dp[j+1][j+2])
    {
        printf("%d %d\n",i,j);
        output(j+1,j+2);
    }
    else if(dp[i][j]==max(a[i],a[j+1])+dp[j][j+2])
    {
        printf("%d %d\n",i,j+1);
        output(j,j+2);
    }
    else
    {
        printf("%d %d\n",j,j+1);
        output(i,j+2);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    printf("%d\n",dfs(1,2));
    output(1,2);
    return 0;
}