[E:10/528, M:8/965, H:0/384] add No: 795 K-th Symbol in Grammar

Author: gladmo

cmd/command/get.go

@@ -11,9 +11,14 @@ import (
 )
 
 var getCmd = &cobra.Command{
-	Use:     "get question_id|leetcode_url",
-	Short:   "get leet question from leetcode-cn.com",
-	Example: "leet get 222",
+	Use:   "get question_id|leetcode_url",
+	Short: "get leetcode question from leetcode-cn.com",
+	Example: `
+leetcode get 795
+leetcode get leetcode-cn.com/problems/k-th-symbol-in-grammar
+leetcode get https://leetcode-cn.com/problems/k-th-symbol-in-grammar
+leetcode get https://leetcode-cn.com/problems/k-th-symbol-in-grammar/solution/
+`,
 	Run: func(cmd *cobra.Command, args []string) {
 
 		if len(args) != 1 {
@@ -33,6 +38,12 @@ var getCmd = &cobra.Command{
 		}
 
 		param := leet.Parse(strings.TrimSpace(args[0]))
+		if param == "" {
+			cmd.Println(fmt.Sprintf("未找到关于 「%s」 的相关题目", args[0]))
+			cmd.Help()
+			os.Exit(1)
+			return
+		}
 
 		res, err := leet.Fetch(param)
 		if err != nil {

cmd/command/info.go

@@ -12,7 +12,7 @@ import (
 
 var infoCmd = &cobra.Command{
 	Use:   "info question_id|leetcode_url",
-	Short: "print leet question info",
+	Short: "print leetcode question info",
 	Run: func(cmd *cobra.Command, args []string) {
 
 		if len(args) != 1 {

leet/leetcode.go

@@ -53,6 +53,11 @@ const (
 	cacheDuration = time.Hour * 24
 )
 
+// Parse ...
+// param: 222
+// param: https://leetcode-cn.com/problems/k-th-symbol-in-grammar
+// param: https://leetcode-cn.com/problems/k-th-symbol-in-grammar/solution/
+// param: leetcode-cn.com/problems/k-th-symbol-in-grammar
 func Parse(param string) string {
 	if match, err := regexp.MatchString(`\d+`, param); err == nil && match {
 		stat, err := ProblemID2name(param)
@@ -61,20 +66,34 @@ func Parse(param string) string {
 		}
 	}
 
+	return ParseFromURL(param)
+}
+
+// ParseFromURL
+// param: https://leetcode-cn.com/problems/k-th-symbol-in-grammar
+// param: https://leetcode-cn.com/problems/k-th-symbol-in-grammar/solution/
+// param: leetcode-cn.com/problems/k-th-symbol-in-grammar
+func ParseFromURL(param string) string {
 	if strings.HasPrefix(param, "http") ||
 		strings.HasPrefix(param, "leetcode-cn.com") {
 
 		re := regexp.MustCompile(`leetcode-cn\.com/problems/(.*)`)
 
 		result := re.FindStringSubmatch(param)
 		if len(result) == 2 {
-			return strings.Trim(result[1], "/")
+			result := strings.Trim(result[1], "/")
+			ss := strings.Split(result, "/")
+			if len(ss) == 2 {
+				result = ss[0]
+			}
+			return result
 		}
 	}
 
 	return ""
 }
 
+// ProblemID2name return problem info
 func ProblemID2name(id string) (stats QuestionStats, err error) {
 	allProblemsDir := path.Join("questions", "all_problems.json")
 	// 文件是否存在 / 结果是否过期

leet/leetcode_test.go

@@ -0,0 +1,66 @@
+package leet
+
+import (
+	"testing"
+)
+
+func TestParseFromURL(t *testing.T) {
+	type args struct {
+		param string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{
+			name: "solution1",
+			args: args{
+				param: "https://leetcode-cn.com/problems/k-th-symbol-in-grammar/solution/",
+			},
+			want: "k-th-symbol-in-grammar",
+		},
+		{
+			name: "solution2",
+			args: args{
+				param: "leetcode-cn.com/problems/k-th-symbol-in-grammar/solution/",
+			},
+			want: "k-th-symbol-in-grammar",
+		},
+		{
+			name: "solution3",
+			args: args{
+				param: "https://leetcode-cn.com/problems/k-th-symbol-in-grammar/",
+			},
+			want: "k-th-symbol-in-grammar",
+		},
+		{
+			name: "solution4",
+			args: args{
+				param: "https://leetcode-cn.com/problems/k-th-symbol-in-grammar",
+			},
+			want: "k-th-symbol-in-grammar",
+		},
+		{
+			name: "solution5",
+			args: args{
+				param: "leetcode-cn.com/problems/k-th-symbol-in-grammar",
+			},
+			want: "k-th-symbol-in-grammar",
+		},
+		{
+			name: "solution6",
+			args: args{
+				param: "leetcode-cn.com/problems/k-th-symbol-in-grammar/",
+			},
+			want: "k-th-symbol-in-grammar",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := ParseFromURL(tt.args.param); got != tt.want {
+				t.Errorf("ParseFromURL() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}

questions/serial/中等/795/golang/main.go

@@ -0,0 +1,90 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"reflect"
+	"time"
+
+	"github.com/gladmo/leetcode/leet"
+	"github.com/gladmo/leetcode/questions/serial/中等/795/golang/solution"
+)
+
+func main() {
+
+	tests := []struct {
+		name   string
+		input1 int
+		input2 int
+		want   int
+	}{
+		{
+			name:   "test-1-1",
+			input1: 1,
+			input2: 1,
+			want:   0,
+		},
+		{
+			name:   "test-2-1",
+			input1: 2,
+			input2: 1,
+			want:   0,
+		},
+		{
+			name:   "test-2-2",
+			input1: 2,
+			input2: 2,
+			want:   1,
+		},
+		{
+			name:   "test-3-2",
+			input1: 3,
+			input2: 2,
+			want:   1,
+		},
+		{
+			name:   "test-3-3",
+			input1: 3,
+			input2: 3,
+			want:   1,
+		},
+		{
+			name:   "test-4-5",
+			input1: 4,
+			input2: 5,
+			want:   1,
+		},
+		{
+			name:   "test-4-7",
+			input1: 4,
+			input2: 7,
+			want:   0,
+		},
+	}
+
+	testLog := leet.NewTestLog(len(tests))
+	defer testLog.Render()
+
+	timeoutDuration := time.Second * 2
+
+	for idx, test := range tests {
+		// 超时检测
+		timeout := leet.Timeout(timeoutDuration, func(ctx context.Context, cancel context.CancelFunc) {
+			solution.Export(test.input1, test.input2)
+			cancel()
+		})
+
+		if timeout {
+			testLog.Fail(idx+1, test.name, "timeout")
+			continue
+		}
+
+		got := solution.Export(test.input1, test.input2)
+		if !reflect.DeepEqual(test.want, got) {
+			testLog.Fail(idx+1, test.name, fmt.Sprintf("want: %v, got %v.", test.want, got))
+			continue
+		}
+
+		testLog.Pass(idx+1, test.name)
+	}
+}

questions/serial/中等/795/golang/solution/k-th-symbol-in-grammar.go

@@ -0,0 +1,14 @@
+package solution
+
+func Export(N int, K int) int {
+	return kthGrammar(N, K)
+}
+
+/****************************************************/
+/******** 以下为 Leetcode 示例部分（提交PR请还原） *******/
+/******** 使用 (./leetcode clear) 初始化所有问题 *******/
+/****************************************************/
+
+func kthGrammar(N int, K int) int {
+
+}

questions/serial/中等/795/golang/solution/k-th-symbol-in-grammar.go.clean

@@ -0,0 +1,14 @@
+package solution
+
+func Export(N int, K int) int {
+	return kthGrammar(N, K)
+}
+
+/****************************************************/
+/******** 以下为 Leetcode 示例部分（提交PR请还原） *******/
+/******** 使用 (./leetcode clear) 初始化所有问题 *******/
+/****************************************************/
+
+func kthGrammar(N int, K int) int {
+
+}

questions/serial/中等/795/question.md

@@ -0,0 +1,31 @@
+## [第K个语法符号](https://leetcode-cn.com/problems/k-th-symbol-in-grammar/)
+
+在第一行我们写上一个 `0`。接下来的每一行，将前一行中的`0`替换为`01`，`1`替换为`10`。
+
+给定行数 `N` 和序数 `K`，返回第 `N` 行中第 `K`个字符。（`K`从1开始）
+
+**例子:**
+
+`**输入:** N = 1, K = 1
+**输出:** 0
+
+**输入:** N = 2, K = 1
+**输出:** 0
+
+**输入:** N = 2, K = 2
+**输出:** 1
+
+**输入:** N = 4, K = 5
+**输出:** 1
+
+**解释:**
+第一行: 0
+第二行: 01
+第三行: 0110
+第四行: 01101001
+`
+
+**注意：**
+
+1.  `N` 的范围 `[1, 30]`.
+2.  `K` 的范围 `[1, 2^(N-1)]`.

questions/store.json

@@ -0,0 +1,90 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"reflect"
+	"time"
+
+	"github.com/gladmo/leetcode/leet"
+	"github.com/gladmo/leetcode/questions/serial/中等/795/golang/solution"
+)
+
+func main() {
+
+	tests := []struct {
+		name   string
+		input1 int
+		input2 int
+		want   int
+	}{
+		{
+			name:   "test-1-1",
+			input1: 1,
+			input2: 1,
+			want:   0,
+		},
+		{
+			name:   "test-2-1",
+			input1: 2,
+			input2: 1,
+			want:   0,
+		},
+		{
+			name:   "test-2-2",
+			input1: 2,
+			input2: 2,
+			want:   1,
+		},
+		{
+			name:   "test-3-2",
+			input1: 3,
+			input2: 2,
+			want:   1,
+		},
+		{
+			name:   "test-3-3",
+			input1: 3,
+			input2: 3,
+			want:   1,
+		},
+		{
+			name:   "test-4-5",
+			input1: 4,
+			input2: 5,
+			want:   1,
+		},
+		{
+			name:   "test-4-7",
+			input1: 4,
+			input2: 7,
+			want:   0,
+		},
+	}
+
+	testLog := leet.NewTestLog(len(tests))
+	defer testLog.Render()
+
+	timeoutDuration := time.Second * 2
+
+	for idx, test := range tests {
+		// 超时检测
+		timeout := leet.Timeout(timeoutDuration, func(ctx context.Context, cancel context.CancelFunc) {
+			solution.Export(test.input1, test.input2)
+			cancel()
+		})
+
+		if timeout {
+			testLog.Fail(idx+1, test.name, "timeout")
+			continue
+		}
+
+		got := solution.Export(test.input1, test.input2)
+		if !reflect.DeepEqual(test.want, got) {
+			testLog.Fail(idx+1, test.name, fmt.Sprintf("want: %v, got %v.", test.want, got))
+			continue
+		}
+
+		testLog.Pass(idx+1, test.name)
+	}
+}