1. 程式人生 > >[HAOI2016]字符合並

[HAOI2016]字符合並

分數 print 枚舉 main efi algorithm ref 轉移 define

Luogu3736

技術分享圖片

很容易想到直接DP,關鍵是枚舉順序。
\(1.\)設後一段構成最後一個點,前一段構成前面的點,那麽能得到\(1\)個點的數量要求 : \(1,k,2k-1...\)相差\(k-1\)
\(2.\)註意循環的正逆順序 : \(mid\)\(j\)小 , 正序枚舉\(j\) ; \(mid\)\(i\)大 , 倒序枚舉\(i\)

具體細節見代碼

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define debug(...) fprintf(stderr,__VA_ARGS__)
#define Debug(x) cout<<#x<<"="<<x<<endl
#define int long long
using namespace std;
typedef long long LL;
const int INF=1e9+7;
inline LL read(){
    register LL x=0,f=1;register char c=getchar();
    while(c<48||c>57){if(c=='-')f=-1;c=getchar();}
    while(c>=48&&c<=57)x=(x<<3)+(x<<1)+(c&15),c=getchar();
    return f*x;
}

int f[305][305][305],g[2],c[305],w[305],a[305];
int n,K,ans=-INF;

signed main(){
    n=read(),K=read();
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=0;i<(1<<K);i++) c[i]=read(),w[i]=read();
    for(int i=1;i<=n;i++){
        for(int j=i;j<=n;j++)
            for(int k=0;k<(1<<K);k++)
                f[i][j][k]=-INF;
    }
    for(int i=n;i>=1;i--)
        for(int j=i;j<=n;j++){
            //註意循環的正逆順序:mid比j小,正序枚舉j;mid比i大,倒序枚舉i
            if(i==j){f[i][j][a[i]]=0;continue;}
            int len=j-i;
            len%=K-1; if(!len) len=K-1;//最終能剩下len個
            for(int mid=j;mid>i;mid-=K-1){
                for(int op=0;op<(1<<len);op++){
                    f[i][j][op<<1]=max(f[i][j][op<<1],f[i][mid-1][op]+f[mid][j][0]);//直接轉移
                    f[i][j][op<<1|1]=max(f[i][j][op<<1|1],f[i][mid-1][op]+f[mid][j][1]);
                }
            }
            if(len==K-1){
                g[0]=g[1]=-INF;
                for(int op=0;op<(1<<K);op++)
                    g[c[op]]=max(g[c[op]],f[i][j][op]+w[op]);//加上操作一次的分數
                f[i][j][0]=g[0],f[i][j][1]=g[1];//用臨時數組更新
            }
        }
    for(int op=0;op<(1<<K);op++)
        ans=max(ans,f[1][n][op]);
    printf("%lld\n",ans);
}

[HAOI2016]字符合並