2020/2021 ICPC 沈阳站 D题 Journey to Un‘Goro （搜索+剪枝）

2020/2021 ICPC 沈阳站 D题 Journey to Un’Goro （搜索+剪枝）

题目链接： 牛客icpc沈阳站复现赛D-Journey to Un’Goro

题意

给定一个整数n (n <= 1e5) 表示一个只由字符r和b构成的字符串序列的长度，对于该序列的任意一个子序列，当该子序列中r的个数为奇数时，则称该子序列为“满意”。要求构造一系列这样的字符串序列使得其中“满意”的子序列个数最大，输出最大个数，另外，这样的序列一般不唯一，题目要求按字典序输出前100个构造的序列。

分析

这里要从序列前缀中r个数的奇偶性上去考虑

设$P_i$表示长度为i的字符序列前缀中r的个数，那么由题意有子序列 (i, j) 中r的个数为$P_j-P_{i-1}$

当且仅当$P_j$与$P_i-1$的奇偶性不同时，$P_j-P_{i-1}$为奇数

那么题目就转化为构造这样的字符串序列，使得序列$P_0,P_1,P_2...P_n$中满足 $P_i$和$P_j$的奇偶性不同的对数最大

（ps: $P_0=0$，这是因为我们假设字符串序列前缀为空时，其中r的个数为0）

设序列$P_0,P_1,P_2...P_n$中奇数个数为$X$, 偶数个数为$Y$

则满足 $P_i$和$P_j$的奇偶性不同的对数为$XY$

通过一些简单的思考很容易想明白$XY=\lfloor (n+1)/2\rfloor \lceil (n+1)/2\rceil$，其实就是$X$和$Y$各取一半时乘积最大，$|X-Y|\le 1$

从上式不难看出，$X$和$Y$都不能超过$\lceil (n+1)/2\rceil$，而同时发现题目只要求输出前100个序列，因此可以搜索+剪枝过掉，在dfs时分别记录当前偶数个数$cn{t}_0$和奇数个数$cn{t}_1$，当 m a x { c n t 0 , c n t 1 } > ⌈ ( n + 1 ) / 2 ⌉ max\{cnt_0,cnt_1\}>\lceil(n+1)/2\rceil max{cnt0​,cnt1​}>⌈(n+1)/2⌉时就剪掉。

由于我们只需要考虑$P_i$的奇偶性，因此搜索策略为枚举当前$P_i$的奇偶性 p i ∈ { 0 , 1 } p_i\in\{0, 1\} pi​∈{0,1}，0表示偶数，1表示奇数，而又因为只有当前位对应为r时才能改变$P_i$的奇偶性，所以当$p_i==p_{i-1}$时，字符串序列上的对应位置b，反之置r。同时考虑需要按字典序输出，因此优先令$p_i==p_{i-1}$

AC代码

#include <bits/stdc++.h>

const int N = 1e5 + 5;

int n, tot, lim;
char S[N];

void dfs(int x, int cnt0, int cnt1, bool st) {
    if(cnt0 > lim || cnt1 > lim) return;
    if(x == n) {
        printf("%s\n", S);
        if(++tot >= 100)
            exit(0);
        return;
    }
    S[x] = 'b';
    dfs(x+1, cnt0 + (st^1), cnt1 + st, st);
    st ^= 1;
    S[x] = 'r';
    dfs(x+1, cnt0 + (st^1), cnt1 + st, st);
}

int main() {
    scanf("%d", &n);
    long long ans = 1ll * (n + 1)/2 * ((n + 2)/2);
    lim = (n+2)/2;
    printf("%lld\n", ans);
    dfs(0, 1, 0, 0);
    return 0;
}