package leetcodejava.array;

import org.junit.Assert;
import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

/**
 * This is the solution of No.1313 problem in the LeetCode,
 * the website of the problem is as follow:
 * https://leetcode-cn.com/problems/decompress-run-length-encoded-list
 * <p>
 * The description of problem is as follow:
 * ==========================================================================================================
 * 给你一个以行程长度编码压缩的整数列表 nums 。
 * <p>
 * 考虑每对相邻的两个元素 [freq, val] = [nums[2*i], nums[2*i+1]] （其中 i >= 0 ），每一对都表示解压后子列表中有 freq 个值为 val 的元素，你需要从左到右连接所有子列表以生成解压后的列表。
 * <p>
 * 请你返回解压后的列表。
 * <p>
 * 示例：
 * <p>
 * 输入：nums = [1,2,3,4]
 * 输出：[2,4,4,4]
 * 解释：第一对 [1,2] 代表着 2 的出现频次为 1，所以生成数组 [2]。
 * 第二对 [3,4] 代表着 4 的出现频次为 3，所以生成数组 [4,4,4]。
 * 最后将它们串联到一起 [2] + [4,4,4] = [2,4,4,4]。
 * 示例 2：
 * <p>
 * 输入：nums = [1,1,2,3]
 * 输出：[1,3,3]
 *  
 * 提示：
 * <p>
 * 2 <= nums.length <= 100
 * nums.length % 2 == 0
 * 1 <= nums[i] <= 100
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/decompress-run-length-encoded-list
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * ==========================================================================================================
 *
 * @author zhangyu (zhangyuyu417@gmail.com)
 */
public class DecompressRunLengthEncoded1313 {
    @Test
    public void decompressRunLengthEncodedTest() {
        int[] nums = {1, 2, 3, 4};
        int[] result = decompressRLElist(nums);
        System.out.println(result);
        Assert.assertEquals(result.length, 4);
    }

    /**
     * 解压链表
     *
     * @param nums 数组
     * @return 解压后数组
     */
    public int[] decompressRLElist(int[] nums) {
        List<Integer> list = new ArrayList<>();
        for (int i = 0; i < nums.length - 1; i += 2) {
            int time = nums[i];
            int num = nums[i + 1];
            for (int j = 0; j < time; j++) {
                list.add(num);
            }
        }
        return list.stream().mapToInt(Integer::valueOf).toArray();
    }
}
