com.javarush.test.level07.lesson09.task04

package com.javarush.test.level07.lesson09.task04;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;

/* Буква «р» и буква «л»
1. Создай список слов, заполни его самостоятельно.
2. Метод fix должен:
2.1. удалять из списка строк все слова, содержащие букву «р»
2.2. удваивать все слова содержащие букву «л».
2.3. если слово содержит и букву «р» и букву «л», то оставить это слово без изменений.
2.4. с другими словами ничего не делать.
Пример:
роза
лира
лоза
Выходные данные:
лира
лоза
лоза
*/

public class Solution
{
    public static void main(String[] args) throws Exception
    {
        ArrayList<String> list = new ArrayList<String>();
        list.add("роза"); //0
        list.add("лира"); //1
        list.add("лоза"); //2
        list.add("тире"); //3
        list.add("путин"); //4
        list.add("уйло"); //5
        list.add("ла-ла"); //6
        list.add("ла-ла"); //7
        list = fix(list);

        for (String s : list)
        {
            System.out.println(s);
        }

    }

    public static ArrayList<String> fix(ArrayList<String> list) {

        String temp;
        for (int i = 0; i < list.size(); i++)
        {
            if (list.get(i).contains("р") && !list.get(i).contains("л")){
                list.remove(i);
                i--;
            }
        }
        for (int i = 0; i < list.size(); i++)
        {
            if (list.get(i).contains("л")&& !list.get(i).contains("р"))
            {
                temp = list.get(i);
                list.add(i + 1, temp);
                i++;
            }
        }
        return list;
    }
}