Monistus

input/code.cpp: In function 'int main()':
input/code.cpp:40:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |                 if (i == stack.size()-1) {
      |                     ~~^~~~~~~~~~~~~~~~~
input/code.cpp:58:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   58 |         if (idx >= input.size()) {
      |             ~~~~^~~~~~~~~~~~~~~

#include <string>
#include <vector>
#include <iostream>
#include <ranges>
 
int main() {
    std::string input;
    std::string output;
 
    std::vector<std::pair<int, int>> stack;
    std::vector<int> count_stack;
 
    std::cin >> input;
 
    int idx = 0;
    int offset = 0;
 
    while (true) {
        /*for (auto& scope : stack) std::cout << scope.first << " " << scope.second << "\n";
 
        std::cout << output << std::endl;
 
        std::cout << input << std::endl;
        for (int i = 0; i < idx; ++i) std::cout << " ";
        std::cout << "^\n";*/
 
        bool loop = true;
 
        int i = stack.size()-1;
 
        while (loop) {
            if (i < 0) {
                break;
            }
 
            if (stack[i].second-offset < idx) {
                int new_idx = stack[i].first;
                loop = false;
 
                if (i == stack.size()-1) {
                    stack.pop_back();
 
                    loop = true;
                    offset = 0;
                } else {
                    // offset += idx-new_idx;
                    offset += stack[i].second - stack[i].first;
 
                    loop = true;
                }
                idx = new_idx;
 
                i = stack.size();
            }
            --i;
        }
 
        if (idx >= input.size()) {
            break;
        }
 
        if (input[idx] >= '1' && input[idx] <= '9') {
            count_stack.push_back(input[idx]-'0');
            stack.push_back(std::pair(idx+1, idx+input[idx] - '0'));
        } else {
            output.push_back(input[idx]);
        }
 
        ++idx;
    }
 
    std::cout << output << std::endl;
 
    return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: WRONG ANSWER

Test 6

Group: 1, 2

Verdict: WRONG ANSWER

Test 7

Group: 1, 2

Verdict: WRONG ANSWER

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED

Test 14

Group: 2

Verdict: ACCEPTED