codeforces 构造 Mark and Lightbulbs

题目:

codeforces 构造 Mark and Lightbulbs_第1张图片

 题目大意:给定一个01起始字符串,和一个目标01字符串,对于一次操作,只可以选择 i 处的某个索引,且 si-1 != si+1。求起始01字符串到目标01字符串的最少操作数量。

算法思路:

        首先,很容易想到的是如果起始字符串和目标字符串的首尾两个字符不相等的话,就肯定没有方法。

        我们看 s[i-1] != s[i+1] , 如果 s[i] 变化了的话,那么 s[i-1] ^ s[i] 会发生改变,s[i] ^ s[i+1]也会发生改变,如果我们构造一个相邻两个数字的异或数组,长度为 n - 1, 如 xor[0] = s[0] ^ s[1], xor[1] = s[1] ^ s[2].......那么就相当于这个异或数组某相邻的两位结果互换,也就是进行一次 0 1 交换。这里引入官方题解的一张图:

codeforces 构造 Mark and Lightbulbs_第2张图片

        然后,如果要想起始字符串变为目标字符串,它们的异或数组中 1 和 0 的个数都是对应相等的,因为没进行一次操作,只是进行一次 0 1 的交换。

        最后,最少的操作数目当然是两个异或数组中对应 1 的位置相减取绝对值,再相加。

        事实上,我们也不需要构造出异或序列来判断,如果 s[i] != s[i+1] 那么异或值为 1 ,直接记录这个索引就行。 

AC代码:

#include
using namespace std;
#include
#define ll long long
#include

void solve() {
    int n; cin >> n;
    string s, t; cin >> s >> t;
    vector pos_s, pos_t;

    if (s[0] != t[0] || s[n - 1] != t[n - 1]) {
        cout << -1 << "\n";
        return;
    }
    for (int i = 0; i < n - 1; i++) {
        if (s[i] != s[i + 1]) pos_s.push_back(i);
        if (t[i] != t[i + 1]) pos_t.push_back(i);
    }
    if (pos_s.size() != pos_t.size()) {
        cout << -1 << "\n";
    }
    else {
        int k = pos_s.size();
        ll ans = 0;
        for (int i = 0; i < k; ++i) {
            ans += abs(pos_s[i] - pos_t[i]);
        }
        cout << ans << "\n";
    }
}

int main() {
    int tt; cin >> tt;
    while (tt--) solve();
}

 

 

 

你可能感兴趣的:(codeforces,构造,codeforces,贪心,算法,c++,贪心算法)