CodeForces - 82D Two out of Three (记忆化搜索+DP)*

题目链接:http://codeforces.com/problemset/problem/82/D

#include<bits/stdc++.h>
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
#define mst(a,b) memset((a),(b),sizeof(a))
const int  maxn =1e3+5;
const int mod=1e9+7;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
题目大意:给定一个序列,
采用一个算法每次选取三个中的两个,权值加上两个中的最大值,
问最后序列的最短时间和打印路径。

先找状态,假设当前进行到i位置,上次选取剩余一个j位置,
那么dp(i,j)就是:从i,i+1,j中再做选择,总共三种扩展,
因为状态数总共只有六次方,所以可以用记忆话搜索。

至于打印路径,直接暴力枚举比较即可,注意对边界条件的判断即可。


*/

int n,a[maxn];///
int dp[maxn][maxn];///dp数组

int Dp(int i,int j){///记忆化搜索
    if(dp[i][j]) return dp[i][j];
    if(i>n) return dp[i][j]=a[j];
    if(i==n) return dp[i][j]=max(a[i],a[j]);
    dp[i][j]=max(a[i],a[j])+Dp(i+2,i+1);
    dp[i][j]=min(dp[i][j],max(a[i],a[i+1])+Dp(i+2,j));
    dp[i][j]=min(dp[i][j],max(a[i+1],a[j])+Dp(i+2,i));
    return dp[i][j];
}

void output(int u,int v){
    if(u==n){
        printf("%d %d\n",v,u);
        return ;
    }
    if(u>n){
        printf("%d\n",v);
        return ;
    }
    int tp=dp[u][v];
    if(tp==max(a[u],a[u+1])+dp[u+2][v]){
        printf("%d %d\n",u,u+1);
        output(u+2,v);
    }
    else if(tp==max(a[u],a[v])+dp[u+2][u+1]){
        printf("%d %d\n",v,u);
        output(u+2,u+1);
    }
    else if(tp==max(a[u+1],a[v])+dp[u+2][u]){
        printf("%d %d\n",v,u+1);
        output(u+2,u);
    }
}


int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    mst(dp,0);///dp肯定是0
    printf("%d\n",Dp(2,1));
    output(2,1);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_37451344/article/details/84033748