Educational DP Contest K-N

K - Stones

题意: K个石子,Taro和Jiro两人轮流从(a1, a2,...,aN)中取ai石子,Taro先取,直到无法取则判负。
思路: dp[i]: i个石子时Taro赢否?(true为赢,false为输),则dp[i] = {dp[i-a[k]]}其中有一个true即为true,dp[0] = false。
代码:

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

const int MAXN = 110;
const int MAXK = 1e5+10;

using ll = long long;

int N, K;
int a[MAXN];


// dp[i]: can i stones win?
int dp[MAXK];

int main() {
    cin >> N >> K;
    for(int i=1; i<=N; i++) cin >> a[i];
    sort(a+1, a+N+1);
    
    dp[0] = 0;
    for(int i=a[0]; i<=K; i++) {
        int flag = 0;
        for(int j=1; j<=N; j++) {
            if(i-a[j]>=0 && !dp[i-a[j]]){
                flag = 1;
                break;
            } 
        }
        if(flag) dp[i] = 1;
    }
    string res = dp[K]?"First":"Second";
    cout << res << endl; 
    return 0;
}

L - Deque

题意: 一堆数,两人轮流从中取(只能头或尾),Taro要最大化X-Y,Jiro要最小化X-Y,求最终结果X-Y。
思路: 有最大有最小,既然要dp显然要调整一下步调。Jiro最小化X-Y,实际上就是最大化Y-X。设dp[i][j]: i到j的X-Y,则每一轮dp[i][j]就是max{Taro(X-Y), Jiro(Y-X)},即dp[i][j] = max{a[i]-dp[i+1][j] , (a[j]-dp[i][j-1])}。
代码

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

using LL = long long;
const int MAXN = 3010;

int A[MAXN];
LL dp[MAXN][MAXN];
int N;

int main() {
    cin >> N;
    for(int i=1; i<=N; i++) cin >> A[i];
    
    memset(dp, 0, sizeof(dp));
    for(int i=N; i>0; i--) {
        for(int j=i; j<=N; j++) {
            if(i==j) dp[i][j] = A[i];
            dp[i][j] = max(A[j]-dp[i][j-1], A[i]-dp[i+1][j]);
        }
    }
    
    cout << dp[1][N] << endl;
    return 0;
}

M - Candies

题意:求n个孩子分m个糖果的种类数,其中第i个孩子最多分a[i]个糖果。
思路:dp[i][j]: 前i个孩子分前j个糖果的和种类数,则有dp[i][j] = \sum_{k=0}^{a[k]}dp[i-1][j-k]
第一种O(n3)方法如下:

#include<bits/stdc++.h>

using namespace std;

const int MAXN  = 110;
const int MAXK = 1e5+10;

int N, K;
int A[MAXN];
int dp[MAXN][MAXK];

int main() {
    cin >> N >> K;
    for(int i=1; i<=N; i++) {
        cin >> A[i];
    }

    dp[0][0] = 1;
    for(int i=1; i<=N; i++) {
        for(int j=0; j<=K; j++) {
            for(int k=0; k<=min(A[i], j); k++) {
                dp[i][j] += dp[i-1][j-k];
            }
        }
    }
    cout << dp[N][K] << endl;
    return 0;
}

但显然超时。
观察到dp[i][j]是从j-min(A[i],j)到j的和,可以用前缀和解决,dp[i][j] = prefix(j) - prefix(j-min(A[i],j)-1),减少一层循环,复杂度为O(n2)。

#include<bits/stdc++.h>

using namespace std;

const int MAXN  = 110;
const int MAXK = 1e5+10;
const int MOD = 1e9+7;

int N, K;
int A[MAXN];
int dp[MAXN][MAXK];
int prefix[MAXK];

int addmod(int a, int b) {
    a += b;
    if(a > MOD) {
        a -= MOD;
    }
    return a;
}

int submod(int a, int b) {
    a -= b;
    if(a < 0) {
        a += MOD;
    }
    return a;
}
int main() {
    cin >> N >> K;
    for(int i=1; i<=N; i++) {
        cin >> A[i];
    }

    dp[0][0] = 1;
    prefix[0] = 1;
    for(int j=1; j<=K; j++) {
        prefix[j] = addmod(prefix[j], prefix[j-1]);
    }
    for(int i=1; i<=N; i++) {
        for(int j=0; j<=K; j++) {
            int L, R;
            L = j-min(A[i],j)-1>=0?prefix[j-min(A[i], j)-1]:0;
            R = prefix[j];
            
            dp[i][j] = addmod(dp[i][j], submod(R, L));
        }
        for(int j=0; j<=K; j++) {
            prefix[j] = dp[i][j];
        }
        for(int j=1; j<=K; j++) {
            prefix[j] = addmod(prefix[j], prefix[j-1]);
        }
    }
    cout << dp[N][K] << endl;
    return 0;
}

N - Slimes

题意:合并石块
思路:区间dp,复杂度O(N3),可以用平行四边形不等式优化,复杂度是O(N2)。
代码:

#include<bits/stdc++.h>

using namespace std;
const int MAXN = 410;
using LL = long long;

int N;
int A[MAXN];
LL dp[MAXN][MAXN];
LL sum[MAXN];

int main() {
    cin >> N;
    sum[0] = 0;
    memset(dp, 0x3f, sizeof(dp));
    for(int i=1; i<=N; i++) {
        cin >> A[i];
        dp[i][i] = 0;
        sum[i] += sum[i-1]+A[i];
    }
    
    for(int len = 1; len<=N-1; len++) {
        for(int i=1; len+i<=N; i++) {
            int j = i+len;
//            if(j>N) continue;
            for(int k=i; k<j; k++) {
                dp[i][j] = min(dp[i][j], dp[i][k]+dp[k+1][j]+sum[j]-sum[i-1]);
            }
        }
    }

    cout << dp[1][N] << endl;
    return 0;
}

平行四边形不等式优化

#include<bits/stdc++.h>

using namespace std;
const int MAXN = 410;
using LL = long long;

int N;
int A[MAXN];
LL dp[MAXN][MAXN];
LL s[MAXN][MAXN];
LL sum[MAXN];

int main() {
    cin >> N;
    sum[0] = 0;
    memset(dp, 0x3f, sizeof(dp));
    for(int i=1; i<=N; i++) {
        cin >> A[i];
        dp[i][i] = 0;
        s[i][i] = i;
        sum[i] += sum[i-1]+A[i];
    }
    
    for(int len = 1; len<=N-1; len++) {
        for(int i=1; len+i<=N; i++) {
            int j = i+len;
//            if(j>N) continue;
            // 平行四边形不等式优化
            for(int k=s[i][j-1]; k<=s[i+1][j]; k++) {
                if(dp[i][k] + dp[k+1][j] + sum[j] - sum[i-1] <= dp[i][j]) {
                    dp[i][j] = dp[i][k] + dp[k+1][j] + sum[j] - sum[i-1];
                    s[i][j] =  k;
                }
            }
        }
    }

    cout << dp[1][N] << endl;
    return 0;
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,558评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,002评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,036评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,024评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,144评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,255评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,295评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,068评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,478评论 1 305
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,789评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,965评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,649评论 4 336
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,267评论 3 318
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,982评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,223评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,800评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,847评论 2 351