Повторить одну строку в столбик, через итератор и цикл (при условии)

197

20 февраля 2019, 22:40

Нужно реализовать повтор строки в столбик 1 раз (если то условие выполняется). Ниже выделил где именно это происходит.

import java.util.*;
import java.lang.*;
public class Solution {
  public static void main(String[] args) throws Exception {
    ArrayList<String> list = new ArrayList<String>();
    list.add("роза");
    list.add("лоза");
    list.add("лира");
    list = fix(list);
    for (String s : list) {
        System.out.println(s);
    }
}
public static ArrayList<String> fix(ArrayList<String> list) {
  for(Iterator<String> it = list.iterator(); it.hasNext();) {
    String nextString = it.next(); 
      if (nextString.contains("р") && !nextString.contains("л")) {
        it.remove();
}
      if (!nextString.contains("р") && nextString.contains("л")) {

// Как тут реализовать повторение строки в столбик - nextString (1 раз) ?

}
      if (nextString.contains("р") && nextString.contains("л")) {
        continue;
     }
   }
        return list;
  }
}

Answer 1

public static void main(String[] args) {
        List<String> list = new ArrayList<String>();
        list.add("роза");
        list.add("лоза");
        list.add("лира");
        list = fix(list);
        for (Iterator<String> it = list.iterator(); it.hasNext(); ) {
            String nextString = it.next();
            System.out.println(nextString);
            it.remove();
        }

    }
    public static List<String> fix(List<String> list) {
        List<String> res = new ArrayList<>();
        for(Iterator<String> it = list.iterator(); it.hasNext();) {
            String nextString = it.next();
            if (!nextString.contains("р") && nextString.contains("л")) {
                System.out.println();
                res.add(nextString);
                res.add(nextString);
            }
        }
        return res;
    }