ABC401D

分类讨论

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;

const int N = 2e5 + 5;
int n, k;
char s[N], t[N];

/*
先预处理一遍字符串 找到锁定位置的.和o,然后如果问号左边是o或者右边是o就把他锁定成.
然后如果剩余o的个数k等于0就把问号锁定成.
如果剩余o的个数k大于0就用一个变量l记录当前连续的问号个数，然后用一个变量mx记录当前连续的o个数，
如果mx等于k就把当前连续的问号个数求出来如果是奇数就把奇数位置锁定成o，把偶数位置锁定成.
如果mx等于k-1就把当前连续的问号个数求如果是偶数就全是？
如果处于两者之间就把所有问号锁定成？
*/
int main()
{
    scanf("%d%d", &n, &k);
    scanf("%s", s + 1);
    for (int i = 1; i <= n; i++)
    {
        if (s[i] == '.')
            t[i] = '.';
        else if (s[i] == 'o')
            t[i] = 'o', k--;
        else
        {
            if (s[i - 1] == 'o' || s[i + 1] == 'o')
            {
                t[i] = s[i] = '.';
            }
        }
    }
    if (k == 0)
    {
        for (int i = 1; i <= n; i++)
        {
            if (s[i] == '?')
                t[i] = '.';
        }
    }
    else
    {
        int mx = 0, l = 0;
        for (int i = 1; i <= n + 1; i++)
        {
            if (s[i] == '?')
                l++;
            else
            {
                mx += (l + 1) / 2;
                l = 0;
            }
        }
        if (mx == k)
        {
            int l = 0;
            for (int i = 1; i <= n + 1; i++)
            {
                if (s[i] == '?')
                    l++;
                else
                {
                    if (l & 1)
                    {
                        for (int j = 1; j <= l; j++)
                            t[i - j] = ".o"[j & 1];
                    }
                    else
                    {
                        for (int j = 1; j <= l; j++)
                            t[i - j] = '?';
                    }
                    l = 0;
                }
            }
        }
        else{
            for(int i = 1; i <= n; i++){
                if(s[i] == '?')
                    t[i] = '?';
            }
        }
    }
    puts(t + 1);
    return 0;
}