leetcode-go/oj/tencent-2017-autumn/encode.go

package main

import "fmt"

func main() {
	var s string
	fmt.Scanln(&s)
	base := []int{16276, 651, 26, 1}
	code, l := 0, len(s)
	num := make([]int, l)
	for i := range num {
		num[i] = int(s[i] - 'a')
	}
	switch l {
	case 1:
		code = num[0] * base[0]
	case 2:
		code = num[0]*base[0] + 1 + num[1]*base[1] + 1
	case 3:
		code = num[0]*base[0] + 1 + num[1]*base[1] + 1 + num[2]*base[2]
	default:
		code = num[0]*base[0] + 1 + num[1]*base[1] + 1 + num[2]*base[2] + 1 + num[3]*base[3]
	}
	fmt.Println(code)
}