1166. 推荐结果打散

1166. 推荐结果打散

 

某个推荐系统会推荐出一个由视频和图片元素组成的列表。x 表示元素编号，视频用 V_x表示，图片用 P_x 表示。现在需要对这些元素进行打散，打散规则如下：

1. 第一个[图片P]出现的位置不变；
2. 从第一个[图片P]后开始，每 n 个元素里面恰好出现1个[图片P]；
3. 图片之间的相对顺序不变；

不能满足打散规则的元素需要舍去。给出元素列表和 n 的值，请返回打散后的元素列表。

样例

 

样例1：

输入样例：

elements=["V_0", "V_1", "V_2", "P_3", "P_4", "P_5", "V_6", "P_7", "V_8", "V_9"] , n=3

输出样例：

["V_0", "V_1", "V_2", "P_3", "V_6", "V_8", "P_4", "V_9"]

样例解释：

P_3 需出现在从0开始的第3位，由于每3个元素中恰好一个 P，P_5 和 P_7 满足不了打散要求，被删除。

 

样例2：

输入样例：

elements=["V_0", "P_1", "V_2", "V_3", "V_4"] , n=2

输出样例：

["V_0", "P_1", "V_2"]

样例解释：

P_1 需出现在从0开始的第1位，由于此后每2个元素中恰好一个 P，V_3 和 V_4 满足不了打散要求，被删除。

注意事项

 

1 ≤元素个数 ≤ 10000

2 ≤ n ≤ 元素个数

保证元素中至少有 1 个 [图片P]

 

 

public class Solution {

    /**

     * @param elements: A list of recommended elements.

     * @param n: [picture P] can appear at most 1 in every n

     * @return: Return the scattered result.

     */

    public List<String> scatter(List<String> elements, int n) {

        // write your code here

       List<String> list = new ArrayList<>();

            int i = 0;

            for (; i < elements.size(); i++) {

                list.add(elements.get(i));

                if (elements.get(i).startsWith("P")) {

                    break;

                }

            }

            List<String> p = new ArrayList<>();

            List<String> v = new ArrayList<>();

            for (; i < elements.size(); i++) {

                if (elements.get(i).startsWith("V")) {

                    v.add(elements.get(i));

                } else {

                    p.add(elements.get(i));

                }

            }

            int flag=1;

            int pI=1;

            for (i=0; i < v.size(); i++) {

                list.add(v.get(i));

                flag++;

                if (flag==n){

                    if(pI==p.size())break;

                    list.add(p.get(pI));

                    pI++;

                    flag=1;

                }

            }

            return list;

    }

}