Codeforces 1370E 二分又二分

最新推荐文章于 2024-09-25 09:08:13 发布

gong_zi_shu

最新推荐文章于 2024-09-25 09:08:13 发布

阅读量98

点赞数

分类专栏：题解文章标签：算法 Codeforces 题目 OI ACM

本文链接：https://blog.csdn.net/weixin_43485513/article/details/107289025

版权

题解专栏收录该内容

46 篇文章 0 订阅

订阅专栏

仔细分析我们可以发现， $s$ 和 $t$ 相同的地方不需要考虑。我们只需要分别记录下不同的地方中， $s$ 为 1 的位置 $s i d$ 以及 $t$ 为 1 的位置 $t i d$ 。

然后我们贪心的想，一步移动中，最多移动的次数，也就是 $t i d$ 的左侧的不同的 $s i d$ 的个数。每一次我们就把这些 $s i d$ 移动到位。然后循环对剩下的 $t i d$ 和 $s i d$ 进行处理， $a n s w e r$ 就是循环了多少次。

由于数据量 $n = 1 e 6$ ，所以必须用二分查找，而且无论是查找 $t i d$ 左侧的 $s i d$ ，还是查找下一个需要查找的 $t i d$ ，都需要二分查找。来保证时间复杂度为 $n l o g n$ 。

代码如下：

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
const int maxn = 1e6 + 5;
char s[maxn], t[maxn];
set<int> sid, tid;
set<int> ::iterator it;
queue<int> de1, de2;
bool visit[maxn];

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    int n;
    cin >> n;
    cin >> s >> t;
    for(int i = 0 ; i < n; i++){
        if(s[i] != t[i]){
            if(s[i] == '1')
                sid.insert(i);
            else
                tid.insert(i);
        }
    }
    if(sid.size() != tid.size())
        cout << -1 << endl;
    else{
        int cnt, ans = 0, temp;
        while(!tid.empty()){
            it = sid.lower_bound(*tid.begin());
            while(it != sid.end()){
                it = tid.lower_bound(*it);
                if(it == tid.end())break;
                temp = *it;
                it = sid.lower_bound(temp);
                it--;
                if(!visit[*it]){
                    visit[*it] = true;
                    de1.push(*it);
                    de2.push(temp);
                }
                it++;
            }
            it = sid.lower_bound(*tid.begin());
            if(it == sid.begin()){
                if(!visit[*sid.rbegin()]){
                    de1.push(*sid.rbegin());
                    de2.push(*tid.begin());
                }
            }else{
                it--;
                de1.push(*it);
                de2.push(*tid.begin());
            }
            while(!de1.empty()){
                sid.erase(de1.front());
                tid.erase(de2.front());
                de1.pop();
                de2.pop();
            }
            ans++;
        }
        cout << ans << endl;
    }
}