The output is correct. How can i fix it? I need to create second list, as i did?
package pl.codegym.task.task07.task0716;
import java.util.ArrayList;
/*
R czy L
*/
public class Solution {
public static void main(String[] args) throws Exception {
ArrayList<String> lista = new ArrayList<String>();
lista.add("rosa"); // 0
lista.add("lira"); // 1
lista.add("luz"); // 2
lista = napraw(lista);
for (String s : lista) {
System.out.println(s);
}
}
public static ArrayList<String> napraw(ArrayList<String> lista) {
ArrayList<String> lista2 = new ArrayList<String>();
lista2.add("rosa");
lista2.add("lira");
lista2.add("luz");
for (int i = 0; i < lista.size(); i++) {
if (lista.get(i).contains("r") && !lista.get(i).contains("l")) {
lista2.remove(i);
} else if (lista.get(i).contains("r") && lista.get(i).contains("l")) {
continue;
} else if (lista.get(i).contains("l") && !lista.get(i).contains("r")) {
lista2.add(lista.get(i));
} else
continue;
}
return lista2;
}
}