✨520.检测大写字母

Author: techoc

README.md

@@ -59,6 +59,7 @@
 - [350.两个数组的交集 II](leetcode/easy/350.go)
 - [387.字符串中的第一个唯一字符](leetcode/easy/387.go)
 - [415.字符串相加](leetcode/easy/t415/solution.go)
+- [520.检测大写字母](leetcode/easy/t520/solution.go)
 - [521.最长特殊序列 Ⅰ](leetcode/easy/521.go)
 - [543.二叉树的直径](leetcode/easy/t543/solution.go)
 - [566.重塑矩阵](leetcode/easy/556.go)

leetcode/easy/t520/solution.go

@@ -0,0 +1,16 @@
+package t520
+
+// 520. 检测大写字母
+// https://leetcode.cn/problems/detect-capital
+func detectCapitalUse(word string) bool {
+	count := 0
+	for _, c := range word {
+		if c >= 'A' && c <= 'Z' {
+			count++
+		}
+	}
+	if count == 0 || count == len(word) || (count == 1 && word[0] >= 'A' && word[0] <= 'Z') {
+		return true
+	}
+	return false
+}

leetcode/easy/t520/solution_test.go

@@ -0,0 +1,30 @@
+package t520
+
+import "testing"
+
+func TestDetectCapitalUse(t *testing.T) {
+	tests := []struct {
+		name  string
+		input string
+		want  bool
+	}{
+		{
+			name:  "Case 1",
+			input: "USA",
+			want:  true,
+		},
+		{
+			name:  "Case 2",
+			input: "FlaG",
+			want:  false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := detectCapitalUse(tt.input); got != tt.want {
+				t.Errorf("detectCapitalUse() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}