[leetcode] 535. Encode and Decode TinyURL @ pythone

原题

Note: This is a companion problem to the System Design problem: Design 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.

解法

构造字符串chars, 然后从中随机取6个字符, 使用random.sample进行不放回抽样, 然后构造字典, 储存longUrl和shortUrl.

代码

class Codec:

    def encode(self, longUrl):
        """Encodes a URL to a shortened URL.
        
        :type longUrl: str
        :rtype: str
        """
        chars = string.ascii_letters + string.digits
        shortUrl = 'http://tinyurl.com/' + ''.join(random.choice(chars) for i in range(6))
        self.data = {}
        self.data[shortUrl] = longUrl
        return shortUrl

    def decode(self, shortUrl):
        """Decodes a shortened URL to its original URL.
        
        :type shortUrl: str
        :rtype: str
        """
        return self.data.get(shortUrl)
        

# Your Codec object will be instantiated and called as such:
# codec = Codec()
# codec.decode(codec.encode(url))