java 代码

public class 气球游戏 {
    /**
     * 使用滑动窗口求解
     * @param args
     */
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        int n=sc.nextInt(),m=sc.nextInt();
        int[] colors=new int[n];
        for (int i=0;i<n;i++) {
            colors[i]=sc.nextInt();
        }
        int res=minWindow(colors,m);
        System.out.println(res);
    }
    private static int minWindow(int[] nums,int m){
        int cnt=m;//判断是否包含所有颜色
        int[] dict=new int[m+1];//记录颜色出现的次数
        for (int i=1;i<=m;i++){
            dict[i]++;
        }
        int begin=0,end=0;
        int width=Integer.MAX_VALUE;
        while (end<nums.length){
            int endNum=nums[end++];
            if (dict[endNum]-->0)
                cnt--;
            while (cnt==0){
                if (end-begin<width)
                    width=end-begin;
                int beginNum=nums[begin++];
                if (dict[beginNum]++==0)
                    cnt++;
            }
        }
        return width==Integer.MAX_VALUE?-1:width;
    }
}