Poistot

input/code.cpp: In function 'void prepJumps(std::vector<int>&, std::vector<int>&)':
input/code.cpp:6:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    6 |     for (int i=0;i<list.size();i++){
      |                  ~^~~~~~~~~~~~
input/code.cpp:9:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    9 |         for (int j=i+1;j<list.size();j++){
      |                        ~^~~~~~~~~~~~
input/code.cpp: In function 'bool round(std::vector<int>&, std::vector<int>&, std::vector<bool>&, std::vector<std::vector<int> >&)':
input/code.cpp:25:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<bool>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   25 |     for (int i=0;i<gone.size();i++){
      |                  ~^~~~~~~~~~~~
i...

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

void prepJumps(vector<int>& list, vector<int>& jumps){
    for (int i=0;i<list.size();i++){
        int n=list[i];
        bool done=false;
        for (int j=i+1;j<list.size();j++){
            if (n<list[j]){
                jumps[i]=j-i;
                done=true;
                break;
            }
        }
        if (!done){
            jumps[i]=-1;
        }
    }
}

bool round(vector<int>& list, vector<int>& jumps, vector<bool>& gone, vector<vector<int>>& removes){
    int starting_index=0;
    bool chosen=false;
    for (int i=0;i<gone.size();i++){
        if (!gone[i]){
            chosen=true;
            starting_index=i;
            break;
        }
    }
    if (!chosen){
        return true;
    }
    removes.push_back({});
    int size=removes.size()-1;
    int index=starting_index;
    //cout<<index<<"\n",
    //gone[starting_index]=true;
    while (jumps[index]!=-1){
        while (gone[index]){
            index+=jumps[index];
            if (jumps[index]=-1){
                if (!gone[index]){
                    gone[index]=true;
                    removes[size].push_back(list[index]);
                    //cout<<list[index]<<" ";
                }
                return false;
            }
        }
        removes[size].push_back(list[index]);
        //cout<<list[index]<<" ";
        gone[index]=true;
        index+=jumps[index];
    }
    if (!gone[index]){
        removes[size].push_back(list[index]);
        //cout<<list[index]<<" ";
        gone[index]=true; 
    }   
    return false;
}

int main(){
    int n=0;
    vector<vector<int>> removes;
    cin>>n;
    vector<int> list(n);
    vector<int> jumps(n);
    vector<bool> gone(n);
    for (int i=0;i<n;i++){
        int k;
        cin>>k;
        list[i]=k;
        gone[i]=false;
    }
    prepJumps(list, jumps);
    bool done=false;
    while (!done){
        done=round(list, jumps, gone, removes);
    }
    cout<<removes.size()<<endl;
    for (vector<int> removed:removes){
        cout<<removed[0];
        for (int k=1;k<removed.size();k++){
            cout<<" "<<removed[k];
        }
        cout<<"\n";
    }
}

Test 1

Group: 1, 3

Verdict: WRONG ANSWER

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 3

Verdict: ACCEPTED

Test 4

Group: 1, 3

Verdict: ACCEPTED

Test 5

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: ACCEPTED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 3

Verdict: TIME LIMIT EXCEEDED