h_nosonの日記

競プロなど

yukicoder No.38 赤青白ブロック

問題
No.38 赤青白ブロック - yukicoder
赤,青,白の各10個のブロックがあり,一列に並べられている.
すべてのブロックに対して
赤の左右Kr個離れた位置に赤がない
青の左右Kb個離れた位置に青がない
を満たすように赤と青のブロックを抜くときの最大のブロックの長さは幾つか

解法
白のブロックは抜く必要がないので計算量は全探索でも2^{20}程度.
作った文字列と何個目のブロックを見ているかを引数に取る関数で深さ優先探索をする.

ソースコード

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

#define RREP(i,s,e) for (i = s; i >= e; i--)
#define rrep(i,n) RREP(i,n,0)
#define REP(i,s,e) for (i = s; i < e; i++)
#define rep(i,n) REP(i,0,n)
#define INF 100000000

typedef long long ll;

int kr, kb;
string s;

int dfs(string ans, int n) {
    if (n == 30)
        return ans.size();
    else if (s[n] == 'R') {
        if (kr > ans.size() || ans[ans.size()-kr] != 'R')
            return max(dfs(ans+s[n],n+1),dfs(ans,n+1));
        else
            return dfs(ans,n+1);
    }
    else if (s[n] == 'B') {
        if (kb > ans.size() || ans[ans.size()-kb] != 'B')
            return max(dfs(ans+s[n],n+1),dfs(ans,n+1));
        else
            return dfs(ans,n+1);
    }
    else
        return dfs(ans+s[n],n+1);
}

int main() {
    cin >> kr >> kb >> s;
    cout << dfs("",0) << endl;
    return 0;
}