mirror of
https://github.com/halfrost/LeetCode-Go.git
synced 2025-07-06 09:23:19 +08:00
283 lines
10 KiB
Go
283 lines
10 KiB
Go
package main
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"strconv"
|
||
"strings"
|
||
)
|
||
|
||
// Graphql define
|
||
type Graphql struct {
|
||
OperationName string `json:"operationName"`
|
||
Variables struct {
|
||
TitleSlug string `json:"titleSlug"`
|
||
} `json:"variables"`
|
||
Query string `json:"query"`
|
||
}
|
||
|
||
// GraphQLResp define
|
||
type GraphQLResp struct {
|
||
Data struct {
|
||
TopicTag TopicTag `json:"topicTag"`
|
||
FavoritesLists FavoritesLists `json:"favoritesLists"`
|
||
} `json:"data"`
|
||
}
|
||
|
||
// TopicTag define
|
||
type TopicTag struct {
|
||
Name string `json:"name"`
|
||
TranslatedName string `json:"translatedName"`
|
||
Slug string `json:"slug"`
|
||
Questions []Question `json:"questions"`
|
||
Frequencies string `json:"frequencies"`
|
||
Typename string `json:"__typename"`
|
||
}
|
||
|
||
// Question define
|
||
type Question struct {
|
||
Status string `json:"status"`
|
||
QuestionID string `json:"questionId"`
|
||
QuestionFrontendID string `json:"questionFrontendId"`
|
||
Title string `json:"title"`
|
||
TitleSlug string `json:"titleSlug"`
|
||
TranslatedTitle string `json:"translatedTitle"`
|
||
Stats string `json:"stats"`
|
||
Difficulty string `json:"difficulty"`
|
||
TopicTags []TopicTags `json:"topicTags"`
|
||
CompanyTags interface{} `json:"companyTags"`
|
||
Typename string `json:"__typename"`
|
||
}
|
||
|
||
// TopicTags define
|
||
type TopicTags struct {
|
||
Name string `json:"name"`
|
||
TranslatedName string `json:"translatedName"`
|
||
Slug string `json:"slug"`
|
||
Typename string `json:"__typename"`
|
||
}
|
||
|
||
func (q Question) generateTagStatus() (TagStatus, error) {
|
||
var ts TagStatus
|
||
err := json.Unmarshal([]byte(q.Stats), &ts)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return ts, err
|
||
}
|
||
return ts, nil
|
||
}
|
||
|
||
// TagStatus define
|
||
type TagStatus struct {
|
||
TotalAccepted string `json:"totalAccepted"`
|
||
TotalSubmission string `json:"totalSubmission"`
|
||
TotalAcceptedRaw int32 `json:"totalAcceptedRaw"`
|
||
TotalSubmissionRaw int32 `json:"totalSubmissionRaw"`
|
||
AcRate string `json:"acRate"`
|
||
}
|
||
|
||
// ConvertMdModelFromQuestions define
|
||
func ConvertMdModelFromQuestions(questions []Question) []Mdrow {
|
||
mdrows := []Mdrow{}
|
||
for _, question := range questions {
|
||
res := Mdrow{}
|
||
v, _ := strconv.Atoi(question.QuestionFrontendID)
|
||
res.FrontendQuestionID = int32(v)
|
||
res.QuestionTitle = strings.TrimSpace(question.Title)
|
||
res.QuestionTitleSlug = strings.TrimSpace(question.TitleSlug)
|
||
q, err := question.generateTagStatus()
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
}
|
||
res.Acceptance = q.AcRate
|
||
res.Difficulty = question.Difficulty
|
||
mdrows = append(mdrows, res)
|
||
}
|
||
return mdrows
|
||
}
|
||
|
||
// TagList define
|
||
type TagList struct {
|
||
FrontendQuestionID int32 `json:"question_id"`
|
||
QuestionTitle string `json:"question__title"`
|
||
SolutionPath string `json:"solution_path"`
|
||
Acceptance string `json:"acceptance"`
|
||
Difficulty string `json:"difficulty"`
|
||
TimeComplexity string `json:"time_complexity"`
|
||
SpaceComplexity string `json:"space_complexity"`
|
||
Favorite string `json:"favorite"`
|
||
}
|
||
|
||
// | 0001 | Two Sum | [Go]({{< relref "/ChapterFour/0001.Two-Sum.md" >}})| Easy | O(n)| O(n)|❤️|50%|
|
||
func (t TagList) tableLine() string {
|
||
return fmt.Sprintf("|%04d|%v|%v|%v|%v|%v|%v|%v|\n", t.FrontendQuestionID, t.QuestionTitle, t.SolutionPath, t.Difficulty, t.TimeComplexity, t.SpaceComplexity, t.Favorite, t.Acceptance)
|
||
}
|
||
|
||
func standardizedTitle(orig string, frontendQuestionID int32) string {
|
||
s0 := strings.TrimSpace(orig)
|
||
s1 := strings.Replace(s0, " ", "-", -1)
|
||
s2 := strings.Replace(s1, "'", "", -1)
|
||
s3 := strings.Replace(s2, "%", "", -1)
|
||
s4 := strings.Replace(s3, "(", "", -1)
|
||
s5 := strings.Replace(s4, ")", "", -1)
|
||
s6 := strings.Replace(s5, ",", "", -1)
|
||
s7 := strings.Replace(s6, "?", "", -1)
|
||
count := 0
|
||
// 去掉 --- 这种情况，这种情况是由于题目标题中包含 - ，左右有空格，左右一填充，造成了 ---，3 个 -
|
||
for i := 0; i < len(s7)-2; i++ {
|
||
if s7[i] == '-' && s7[i+1] == '-' && s7[i+2] == '-' {
|
||
fmt.Printf("【需要修正 --- 的标题是 %v】\n", fmt.Sprintf("%04d.%v", int(frontendQuestionID), s7))
|
||
s7 = s7[:i+1] + s7[i+3:]
|
||
count++
|
||
}
|
||
}
|
||
if count > 0 {
|
||
fmt.Printf("总共修正了 %v 个标题\n", count)
|
||
}
|
||
// 去掉 -- 这种情况，这种情况是由于题目标题中包含负号 -
|
||
for i := 0; i < len(s7)-2; i++ {
|
||
if s7[i] == '-' && s7[i+1] == '-' {
|
||
fmt.Printf("【需要修正 -- 的标题是 %v】\n", fmt.Sprintf("%04d.%v", int(frontendQuestionID), s7))
|
||
s7 = s7[:i+1] + s7[i+2:]
|
||
count++
|
||
}
|
||
}
|
||
if count > 0 {
|
||
fmt.Printf("总共修正了 %v 个标题\n", count)
|
||
}
|
||
return s7
|
||
}
|
||
|
||
// GenerateTagMdRows define
|
||
func GenerateTagMdRows(solutionIds []int, metaMap map[int]TagList, mdrows []Mdrow, internal bool) []TagList {
|
||
tl := []TagList{}
|
||
for _, row := range mdrows {
|
||
if BinarySearch(solutionIds, int(row.FrontendQuestionID)) != -1 {
|
||
tmp := TagList{}
|
||
tmp.FrontendQuestionID = row.FrontendQuestionID
|
||
tmp.QuestionTitle = strings.TrimSpace(row.QuestionTitle)
|
||
s7 := standardizedTitle(row.QuestionTitle, row.FrontendQuestionID)
|
||
if internal {
|
||
tmp.SolutionPath = fmt.Sprintf("[Go]({{< relref \"/ChapterFour/%v/%v.md\" >}})", GetChpaterFourFileNum(int(row.FrontendQuestionID)), fmt.Sprintf("%04d.%v", int(row.FrontendQuestionID), s7))
|
||
} else {
|
||
tmp.SolutionPath = fmt.Sprintf("[Go](https://books.halfrost.com/leetcode/ChapterFour/%v/%v)", GetChpaterFourFileNum(int(row.FrontendQuestionID)), fmt.Sprintf("%04d.%v", int(row.FrontendQuestionID), s7))
|
||
}
|
||
tmp.Acceptance = row.Acceptance
|
||
tmp.Difficulty = row.Difficulty
|
||
tmp.TimeComplexity = metaMap[int(row.FrontendQuestionID)].TimeComplexity
|
||
tmp.SpaceComplexity = metaMap[int(row.FrontendQuestionID)].SpaceComplexity
|
||
tmp.Favorite = metaMap[int(row.FrontendQuestionID)].Favorite
|
||
tl = append(tl, tmp)
|
||
}
|
||
}
|
||
return tl
|
||
}
|
||
|
||
// TagLists define
|
||
type TagLists struct {
|
||
TagLists []TagList
|
||
}
|
||
|
||
// | No. | Title | Solution | Difficulty | TimeComplexity | SpaceComplexity |Favorite| Acceptance |
|
||
// |:--------:|:------- | :--------: | :----------: | :----: | :-----: | :-----: |:-----: |
|
||
func (tls TagLists) table() string {
|
||
res := "| No. | Title | Solution | Difficulty | TimeComplexity | SpaceComplexity |Favorite| Acceptance |\n"
|
||
res += "|:--------:|:------- | :--------: | :----------: | :----: | :-----: | :-----: |:-----: |\n"
|
||
for _, p := range tls.TagLists {
|
||
res += p.tableLine()
|
||
}
|
||
// 加这一行是为了撑开整个表格
|
||
res += "|------------|-------------------------------------------------------|-------| ----------------| ---------------|-------------|-------------|-------------|"
|
||
return res
|
||
}
|
||
|
||
// AvailableTagTable define
|
||
func (tls TagLists) AvailableTagTable() string {
|
||
return tls.table()
|
||
}
|
||
|
||
// FavoritesLists define
|
||
type FavoritesLists struct {
|
||
PublicFavorites []int `json:"publicFavorites"`
|
||
PrivateFavorites []struct {
|
||
IDHash string `json:"idHash"`
|
||
ID string `json:"id"`
|
||
Name string `json:"name"`
|
||
IsPublicFavorite bool `json:"isPublicFavorite"`
|
||
ViewCount int `json:"viewCount"`
|
||
Creator string `json:"creator"`
|
||
IsWatched bool `json:"isWatched"`
|
||
Questions []struct {
|
||
QuestionID string `json:"questionId"`
|
||
Title string `json:"title"`
|
||
TitleSlug string `json:"titleSlug"`
|
||
Typename string `json:"__typename"`
|
||
} `json:"questions"`
|
||
Typename string `json:"__typename"`
|
||
} `json:"privateFavorites"`
|
||
Typename string `json:"__typename"`
|
||
}
|
||
|
||
// Gproblem define
|
||
type Gproblem struct {
|
||
QuestionID string `json:"questionId"`
|
||
QuestionFrontendID string `json:"questionFrontendId"`
|
||
BoundTopicID int `json:"boundTopicId"`
|
||
Title string `json:"title"`
|
||
TitleSlug string `json:"titleSlug"`
|
||
Content string `json:"content"`
|
||
TranslatedTitle string `json:"translatedTitle"`
|
||
TranslatedContent string `json:"translatedContent"`
|
||
IsPaidOnly bool `json:"isPaidOnly"`
|
||
Difficulty string `json:"difficulty"`
|
||
Likes int `json:"likes"`
|
||
Dislikes int `json:"dislikes"`
|
||
IsLiked interface{} `json:"isLiked"`
|
||
SimilarQuestions string `json:"similarQuestions"`
|
||
Contributors []interface{} `json:"contributors"`
|
||
LangToValidPlayground string `json:"langToValidPlayground"`
|
||
TopicTags []struct {
|
||
Name string `json:"name"`
|
||
Slug string `json:"slug"`
|
||
TranslatedName string `json:"translatedName"`
|
||
Typename string `json:"__typename"`
|
||
} `json:"topicTags"`
|
||
CompanyTagStats interface{} `json:"companyTagStats"`
|
||
CodeSnippets []GcodeSnippet `json:"codeSnippets"`
|
||
Stats string `json:"stats"`
|
||
Hints []interface{} `json:"hints"`
|
||
Solution interface{} `json:"solution"`
|
||
Status interface{} `json:"status"`
|
||
SampleTestCase string `json:"sampleTestCase"`
|
||
MetaData string `json:"metaData"`
|
||
JudgerAvailable bool `json:"judgerAvailable"`
|
||
JudgeType string `json:"judgeType"`
|
||
MysqlSchemas []interface{} `json:"mysqlSchemas"`
|
||
EnableRunCode bool `json:"enableRunCode"`
|
||
EnableTestMode bool `json:"enableTestMode"`
|
||
EnvInfo string `json:"envInfo"`
|
||
Typename string `json:"__typename"`
|
||
}
|
||
|
||
// Gstat define
|
||
type Gstat struct {
|
||
TotalAcs int `json:"total_acs"`
|
||
QuestionTitle string `json:"question__title"`
|
||
IsNewQuestion bool `json:"is_new_question"`
|
||
QuestionArticleSlug string `json:"question__article__slug"`
|
||
TotalSubmitted int `json:"total_submitted"`
|
||
FrontendQuestionID int `json:"frontend_question_id"`
|
||
QuestionTitleSlug string `json:"question__title_slug"`
|
||
QuestionArticleLive bool `json:"question__article__live"`
|
||
QuestionHide bool `json:"question__hide"`
|
||
QuestionID int `json:"question_id"`
|
||
}
|
||
|
||
// GcodeSnippet define
|
||
type GcodeSnippet struct {
|
||
Lang string `json:"lang"`
|
||
LangSlug string `json:"langSlug"`
|
||
Code string `json:"code"`
|
||
Typename string `json:"__typename"`
|
||
}
|