НП - Лаб 5-1 Супер-стринг

/*
Треба да се напише класа SuperString. Класата во позадина претставува листа на стрингови LinkedList<String> и ги нуди следните методи:

SuperString() - креира празен стринг
append(String s) - го додава стрингот на крајот во листата
insert(String s) - го додава стрингот на почеток на листата
contains(String s):boolean - враќа true доколку стрингот s се наоѓа во супер-стрингот. Стрингот s може да е разделен во повеќе подстрингови во листата. Пр: list = [ "st" , "arz" , "andrej" ] , contains("tarzan") –> true
reverse() - го превртува стрингот на следниов начин. Ги превртува сите елементи во листата, а потоа и секој подстринг како елемент посебно го превртува. list = [ "st" , "arz" , "andrej: ]; reverse(); list = [ "jerdna", "zra", "ts"]
toString():String - ги враќа конкатенирани сите елементи во листата list = [ "st" , "arz" , "andrej"]; toString() -> "starzandrej"
removeLast(int k) – ги отстранува последнo додадените k подстрингови
*/

import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Scanner;
import java.util.stream.Collectors;

class SuperString {
    LinkedList<String> lista;
    List<String> lastAdded;

    public SuperString() {
        lista = new LinkedList<>();
        lastAdded = new ArrayList<>();
    }
    public void append(String s) {
        lista.add(s);
        lastAdded.add(0,s);
    }
    public void insert(String s) {
        lista.add(0, s);
        lastAdded.add(0,s);
    }
    public boolean contains(String s) {
        return this.toString().contains(s);
    }
    public void reverse() {
        Collections.reverse(lista);
        for (int i=0; i<lista.size(); i++) {
            lista.set(i, reverseString(lista.get(i)));
            lastAdded.set(i, reverseString(lastAdded.get(i)));
        }
    }
    public void removeLast(int k) {
        for(String s : lastAdded.subList(0, k))
            lista.remove(s);
        lastAdded = lastAdded.subList(k, lastAdded.size());
    }
    @Override
    public String toString() {
        return lista.stream().collect(Collectors.joining());
    }
    public String reverseString(String s) {
        String reverse="";
        for(int i = s.length()-1; i>=0; i--)
            reverse += s.charAt(i);
        return reverse;
    }
}

public class SuperStringTest {

    public static void main(String[] args) {
        Scanner jin = new Scanner(System.in);
        int k = jin.nextInt();
        if (  k == 0 ) {
            SuperString s = new SuperString();
            while ( true ) {
                int command = jin.nextInt();
                if ( command == 0 ) {//append(String s)
                    s.append(jin.next());
                }
                if ( command == 1 ) {//insert(String s)
                    s.insert(jin.next());
                }
                if ( command == 2 ) {//contains(String s)
                    System.out.println(s.contains(jin.next()));
                }
                if ( command == 3 ) {//reverse()
                    s.reverse();
                }
                if ( command == 4 ) {//toString()
                    System.out.println(s);
                }
                if ( command == 5 ) {//removeLast(int k)
                    s.removeLast(jin.nextInt());
                }
                if ( command == 6 ) {//end
                    break;
                }
            }
        }
    }

}