Code is giving correct result however not passing the verification. Need Help!
import java.util.ArrayList;
import java.util.ListIterator;
/*
R or L
*/
public class Solution {
public static void main(String[] args) throws Exception {
ArrayList<String> list = new ArrayList<String>();
list.add("rose"); //0
list.add("love"); //1
list.add("lyre"); //2
list = fix(list);
for (String s : list) {
System.out.println(s);
}
}
public static ArrayList<String> fix(ArrayList<String> list) {
int i = 0;
String[] arr = new String[list.size()];
ListIterator<String> itr = list.listIterator();
while(itr.hasNext())
{
String s = itr.next();
if( s.contains("r") && !s.contains("l") ) // only "r"
{
itr.remove();
}
else if( s.contains("l") && !s.contains("r") ) // only "l"
{
arr[i] = s;
i++;
}
}
for(i = 0; arr[i] != null; i++)
{
itr.add(arr[i]);
}
return list;
}
}