codeforces Round #502 (in memory of Leopoldo Taravilse, Div. 1 + Div. 2)---B

        满足交换了有变化的情况有s[i] = 0 t[i] = 0 和 s[i] = 1 t[i] = 0 的情况,那么我们先统计s串里面有多少个1和0,因为第一种和所有的1交换都可以,第二种和所有的0交换都可以改变xor之后的结果。

       那么对每一个第一种情况,答案加上1的个数,第二种情况答案加上0的个数,但是因为满足上面两种情况的位置的0和1可以相互交换,因此存在一些重复的情况,我们统计出所有上面的情况,那么满足情况的可能多余的数据就是情况数为k的话,就是(k-1)*k/2 但是 因为0和0 1和1 是不能交换的,因此需要加上情况数,如果满足情况里面第一种的个数 k1 那么加上(k1-1)*k1/2 满足第二种k2同理,

代码如下

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
int main(){
    int n;
    cin >> n;
    string s1,s2;
    cin >> s1 >> s2;
    LL cnt1 = 0,cnt2 = 0;
    for (int i=0; i<n; i++) {
        if (s1[i] == '0') cnt1++;
        else cnt2++;
    }
    LL ans = 0,cnt11 = 0,cnt22 = 0;
    for (int i=0; i<n; i++) {
        if (s1[i] == '0' && s2[i] == '0'){
            ans += cnt2;
            cnt11++;
        }else if (s1[i] == '1' && s2[i] == '0'){
            ans += cnt1;
            cnt22++;
        }
    }
    LL cnt = cnt11 + cnt22;
    if (cnt > 0) ans -= (cnt-1)*cnt/2;
    if (cnt11 > 0) ans += (cnt11-1)*cnt11/2;
    if (cnt22 > 0) ans += (cnt22-1)*cnt22/2;
    cout << ans << endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/CCCCTong/article/details/81590463