MarsCode刷题--74. 叠盘子排序

问题描述

小M有一个独特的方式来收拾家中的盘子。每次用餐后，他会将盘子按照他们的序号顺序叠放。盘子的序号都是唯一的整数，并且在收拾前就是递增的。小M的叠放规则是，每一堆盘子的序号都是连续递增的，并且至少包含3个盘子。需要编写程序帮助小M确定盘子的叠放方式。

例如，输入的盘子序号是 [-3, -2, -1, 2, 10, 15, 16, 18, 19, 20]，按照小M的规则，连续递增序列 -3, -2, -1 可以叠在一起表示为 -3–1，而 18, 19, 20 可以叠在一起表示为 18-20。不满足连续递增至少3个的，如 2, 10, 15, 16 都应单独列出。

输入参数

plates: 一个整数数组，表示盘子的序号
n: 盘子总数

测试样例

样例1：

输入：plates = [-3, -2, -1, 2, 10, 15, 16, 18, 19, 20], n = 10
输出："-3--1,2,10,15,16,18-20"

样例2：

输入：plates = [-6, -3, -2, -1, 0, 1, 3, 4, 5, 7, 8, 9, 10, 11, 14, 15, 17, 18, 19, 20], n = 20
输出："-6,-3-1,3-5,7-11,14,15,17-20"

样例3：

输入：plates = [1, 2, 7, 8, 9, 10, 11, 19], n = 8
输出："1,2,7-11,19"

解题思路

Java代码

public class Main {
    public static String solution(int[] plates, int n) {
        // Please write your code here
        int length = plates.length;
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < length; i++) {
            int start = i;
            while (i + 1 < length && plates[i + 1] - plates[i] == 1) {
                i++;
            }
            if (i - start + 1 >= 3) {
                sb.append(plates[start]).append("-").append(plates[i]).append(",");
            } else if (i - start + 1 == 2) {
                sb.append(plates[start]).append(",").append(plates[start + 1]).append(",");
            } else {
                sb.append(plates[start]).append(",");
            }
        }
        sb.deleteCharAt(sb.length() - 1);
        return sb.toString();
    }

    public static void main(String[] args) {
        // You can add more test cases here
        System.out.println(
                solution(new int[] { -3, -2, -1, 2, 10, 15, 16, 18, 19, 20 }, 10).equals("-3--1,2,10,15,16,18-20"));
        System.out.println(
                solution(new int[] { -6, -3, -2, -1, 0, 1, 3, 4, 5, 7, 8, 9, 10, 11, 14, 15, 17, 18, 19, 20 }, 20)
                        .equals("-6,-3-1,3-5,7-11,14,15,17-20"));
        System.out.println(solution(new int[] { 1, 2, 7, 8, 9, 10, 11, 19 }, 8).equals("1,2,7-11,19"));
    }
}