LeetCode - Encode And Decode TinyURL

题目

TinyURL is a URL shortening service where you enter a URL such as https://leetcode.com/problems/design-tinyurl and it returns a short URL such as http://tinyurl.com/4e9iAk.

Design the encode and decode methods for the TinyURL service. There is no restriction on how your encode/decode algorithm should work. You just need to ensure that a URL can be encoded to a tiny URL and the tiny URL can be decoded to the original URL.

　　输入一个长的URL，输出一个短的URL，并且每次输入输出要对应。

思路

　　将长URL和短URL在map中双向映射，生成短URL时随机选取6个字符，如果有重复的，就再次生成。

代码

public class EncodeAndDecodeTinyURL {

    private static final String CODE = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";

    private Map<String, String> map = new HashMap<>();

    // Encodes a URL to a shortened URL.
    public String encode(String longUrl) {
        String res = map.get(longUrl);
        if (res != null) {
            return res;
        }
        res = "http://tinyurl.com/";
        Random random = new Random();
        for (int i = 0; i < 6; i++) {
            res += CODE.charAt(random.nextInt(62));
        }
        // 有重复的就再次生成
        while (map.containsKey(res)) {
            for (int i = 0; i < 6; i++) {
                res += CODE.charAt(random.nextInt(62));
            }
        }
        map.put(longUrl, res);
        map.put(res, longUrl);
        return res;
    }

    // Decodes a shortened URL to its original URL.
    public String decode(String shortUrl) {
        return map.get(shortUrl);
    }

}