leetcode/2379.cpp at master · Yashbhatt7/leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include<iostream>
#include<vector>
#include<algorithm>

void crt_vec(std::string &temp, int count, int op_cnt, std::vector<int> &op_count, int k, int itr) {
    /*std::cout << itr << "\n";*/

    for(int i = itr; i < temp.size(); ++i) {
        if(temp[i] == 'B') {
            ++count;
        }
        else {
            temp[i] = 'B';
            ++count;
            ++op_cnt;
        }

        if(count >= k) {
            std::cout << count << "\n";
            op_count.push_back(op_cnt);
            count = 0;
            op_cnt = 0;
            break;
        }
    }
}

/*std::vector<int> minimumRecolors(std::string block, int k) {*/
int minimumRecolors(std::string block, int k) {
    int count = 0;
    int op_cnt = 0;
    std::string temp = block;
    std::string rm_temp = block;
    std::vector<int> op_count;
    int min = 0;

    for(int i = 0; i < block.size(); ++i) {
        if(block[i] == 'B') {
            ++count;
            if(count >= k) {
                std::cout << "\n\n\nadsfasdf\n\n\n";
                return 0;
                /*return op_count;*/
            }
        }
        else {
            count = 0;
        }
    }
    count = 0;
    op_cnt = 0;

    if(block[0] == 'B') {
        crt_vec(temp, count, op_cnt, op_count, k, 0);
        temp = block;
    }

    int last_w_index = block.rfind('W');
    for(int i = 0; i < block.size(); ++i) {
        crt_vec(temp, count, op_cnt, op_count, k, i);
        temp = block;

        if(i == last_w_index) {
            break;
        }
    }

    int res = 0;
    if(!op_count.empty()) {
        auto min_iter = std::min_element(op_count.begin(), op_count.end());
        res = *min_iter;
    }

    return res;
    /*return op_count;*/
}

int main() {
    std::string block = "WWBBBWBBBBBWWBWWWB";
    /*std::string block = "WBBWWWWBBWWBBBBWWBBWWBBBWWBBBWWWBWBWW";*/
    /*std::string block = "BWWWBB";*/
    int k = 16;
    /*int k = 6;*/
    /*int k = 15;*/

    std::cout << minimumRecolors(block, k);

    /*std::vector<int> res = minimumRecolors(block, k);*/
    /*for(int x : res) {*/
    /*    std::cout << x << " ";*/
    /*}*/

    std::cin.get();
}