ACM-ICPC 2018 南京赛区现场赛 E. Eva and Euro coins （思维）

题目链接：https://codeforc.es/gym/101981/attachments

题意：给出两个只包含01的字符串，每次可以选择连续k个相同的数字进行翻转，问能否通过若干次操作把两个字符串变为相同。

题解：（qls：通过观察可以发现，可以把每个 1 在不跨越其他 1 的情况下往左/右移 k 个位置，尽可能把 1 往左移，出现连续 k 个 1 就消掉，check一下两个串操作完之后是否相等。）

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define mst(a,b) memset((a),(b),sizeof(a))
#define mp(a,b) make_pair(a,b)
#define pi acos(-1)
#define pii pair<int,int>
#define pb push_back
const int INF = 0x3f3f3f3f;
const double eps = 1e-6;
const int maxn = 1e6 + 10;
const int maxm = 1e5 + 10;
const ll mod =  998244353;

int n,k;
string s1,s2;
int st[maxn],cnt[maxn];

string F(string s) {
    if(k == 1) return "";
    int top = 0;
    for(int i = 0; i < n; i++) {
        if(top && st[top] == s[i] - '0') {
            cnt[top]++;
            if(cnt[top] == k) top--;
        } else {
            st[++top] = s[i] - '0';
            cnt[top] = 1;
        }
    }
    s = "";
    for(int i = 1; i <= top; i++) {
        while(cnt[i]) {
            s += st[i];
            cnt[i]--;
        }
    }
    return s;
}

int main() {
#ifdef local
    freopen("data.txt", "r", stdin);
//    freopen("data.txt", "w", stdout);
#endif
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> k >> s1 >> s2;
    if(F(s1) == F(s2)) cout << "Yes" << endl;
    else cout << "No" << endl;
    return 0;
}