UVA11127- Triple-Free Binary Strings(DFS+位运算)

题目链接

题意：给出长度为n的字符串，字符串由‘1’，‘0’，‘*’组成，其中‘*’可以任意替换为‘1’，‘0’，求不存在连续3个相同子串的字符串的最多个数。

思路：我们可以利用二进制的形式来表示字符串，进行DFS。利用位运算的左移来表示在‘*’位置上放置‘1’，注意在递归的过程中注意判断之否存在3个连续相同的子串。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int MAXN = 32;

char str[MAXN];
int n, ans;

bool judge(int s, int d) {
    int k = (1 << d) - 1;
    int a = s & k;
    s = s >> d;
    int b = s & k;
    s = s >> d;
    int c = s & k;
    if (a == b && b == c)
        return true;
    else
        return false;
}//利用位运算，右移d位，判断是否存在3个子串相同

void dfs(int s, int cnt) {
    int num = cnt / 3;  
    int t = n - cnt - 1;
    for (int i = 1; i <= num; i++) {
        if (judge(s >> (t + 1), i)) 
            return;
    }
    if (cnt == n) {
        ans++;
        return;
    }
    else if (str[cnt] == '0') {
         dfs(s, cnt + 1);
    }
    else if (str[cnt] == '1') {
         dfs(s + (1 << t), cnt + 1);
    }
    else if (str[cnt] == '*') {
         dfs(s, cnt + 1);
         dfs(s + (1 << t), cnt + 1);  
    }
    return;
}

int main() {
    int cas = 1;
    while (scanf("%d", &n) && n) {
        scanf("%s", str);
        ans = 0;
        dfs(0, 0); 
        printf("Case %d: %d\n", cas++, ans);
    }    
    return 0;
}