leetcode 403. 青蛙过河

一只青蛙想要过河。 假定河流被等分为 x 个单元格，并且在每一个单元格内都有可能放有一石子（也有可能没有）。 青蛙可以跳上石头，但是不可以跳入水中。
给定石子的位置列表（用单元格序号升序表示）， 请判定青蛙能否成功过河（即能否在最后一步跳至最后一个石子上）。 开始时， 青蛙默认已站在第一个石子上，并可以假定它第一步只能跳跃一个单位（即只能从单元格1跳至单元格2）。
如果青蛙上一步跳跃了 k 个单位，那么它接下来的跳跃距离只能选择为 k - 1、k 或 k + 1个单位。 另请注意，青蛙只能向前方（终点的方向）跳跃。
请注意：
石子的数量 ≥ 2 且 < 1100；
每一个石子的位置序号都是一个非负整数，且其 < 231；
第一个石子的位置永远是0。

思路：利用Map套Set存能到达到i的所有k，遍历一遍即可。

public class __403 {
    Map<Integer,Set<Integer>> map;
    public boolean canCross(int[] stones) {
        if(stones.length==0){
            return false;
        }
        map=new HashMap<Integer, Set<Integer>>();
        for(int i=0;i<stones.length;++i){
            map.put(stones[i],new HashSet<Integer>());
        }
        map.get(0).add(0);
        for(int i=0;i<stones.length;++i){
            for(int k:map.get(stones[i])){
                for(int j=k-1;j<=k+1;++j){
                    if(j>0&&map.containsKey(stones[i]+j)){
                        map.get(stones[i]+j).add(j);
                    }
                }
            }
        }
        return map.get(stones[stones.length-1]).size()>0;
    }
    public static void main(String[] args) {
        int[] a={0,1,3,4,5,7,9,10,12};
        System.out.println(new __403().canCross(a));
    }
}