<添加><算法><宝石包涵>

Author: iceopen

leetcode/jewels-and-stones/jewels-and-stones.go

@@ -0,0 +1,43 @@
+package main
+
+import (
+	"fmt"
+)
+
+// 给定字符串J 代表石头中宝石的类型，和字符串 S代表你拥有的石头。 S 中每个字符代表了一种你拥有的石头的类型，你想知道你拥有的石头中有多少是宝石。
+//
+// J 中的字母不重复，J 和 S中的所有字符都是字母。字母区分大小写，因此"a"和"A"是不同类型的石头。
+//
+// 示例 1:
+//
+// 输入: J = "aA", S = "aAAbbbb"
+// 输出: 3
+// 示例 2:
+//
+// 输入: J = "z", S = "ZZ"
+// 输出: 0
+// 注意:
+//
+// S 和 J 最多含有50个字母。
+// J 中的字符不重复。
+
+func numJewelsInStones(J string, S string) int {
+	m := make(map[byte]int)
+
+	for _, v := range S {
+		m[byte(v)]++
+	}
+
+	sum := 0
+	for _, v := range J {
+		sum += m[byte(v)]
+	}
+	return sum
+}
+
+func main() {
+	J := "aA"
+	S := "aAAbbbb"
+	i := numJewelsInStones(J, S)
+	fmt.Println("宝石数量：", i)
+}

leetcode/median-of-two-sorted-arrays/median-of-two-sorted-arrays.go

@@ -1,29 +1,116 @@
 package main
 
+import (
+	"fmt"
+)
 
-// 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
+// 给定两个大小为 m 和 n 的有序数组 A 和 B 。
 //
 // 请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。
 //
-// 你可以假设 nums1 和 nums2 不同时为空。
+// 你可以假设 A 和 B 不同时为空。
 //
 // 示例 1:
 //
-// nums1 = [1, 3]
-// nums2 = [2]
+// A = [1, 3]
+// B = [2]
 //
 // 中位数是 2.0
 // 示例 2:
 //
-// nums1 = [1, 2]
-// nums2 = [3, 4]
+// A = [1, 2]
+// B = [3, 4]
 //
 // 中位数是 (2 + 3)/2 = 2.5
 
 func main() {
+	f := findMedianSortedArrays([]int{1, 2}, []int{3, 4})
+	fmt.Println(f)
+}
+// 简单方法 TODO 测试效率不高 有时间压测下
+// func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
+// 	var nums []int
+// 	var f float64
+// 	for i := 0; i < len(nums2); i++ {
+// 		nums1 = append(nums1, nums2[i])
+// 	}
+// 	nums = nums1
+// 	sort.Ints(nums)
+// 	if len(nums)%2 != 0 {
+// 		index := float64(len(nums)) / 2
+// 		index = math.Floor(index + 0.5)
+// 		f = float64(nums[int(index)-1])
+// 	} else {
+// 		index := len(nums) / 2
+// 		f = float64(nums[index]+nums[index-1]) / 2
+// 	}
+// 	return f
+// }
+
+func findMedianSortedArrays(A []int, B []int) float64 {
+	m := len(A)
+	n := len(B)
+
+	if m > n { // to ensure A <= B
+		temp := A
+		A = B
+		B = temp
+
+		tmp := m
+		m = n
+		n = tmp
+	}
+	iMin := 0
+	iMax := m
+	halfLen := (m + n + 1) / 2
+	for {
+		if iMin > iMax {
+			break
+		}
+		i := (iMin + iMax) / 2
+		j := halfLen - i
+		if i < iMax && B[j-1] > A[i] {
+			iMin = i + 1 // i is too small
+		} else if i > iMin && A[i-1] > B[j] {
+			iMax = i - 1 // i is too big
+		} else { // i is perfect
+			maxLeft := 0
+			if i == 0 {
+				maxLeft = B[j-1]
+			} else if j == 0 {
+				maxLeft = A[i-1]
+			} else {
+				maxLeft = Max(A[i-1], B[j-1])
+			}
+			if (m+n)%2 == 1 {
+				return float64(maxLeft)
+			}
 
+			minRight := 0
+			if i == m {
+				minRight = B[j]
+			} else if j == n {
+				minRight = A[i]
+			} else {
+				minRight = Min(B[j], A[i])
+			}
+
+			return float64(maxLeft+minRight) / 2.0
+		}
+	}
+	return 0
 }
 
-func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
+func Max(x, y int) int {
+	if x > y {
+		return x
+	}
+	return y
+}
 
-}
+func Min(x, y int) int {
+	if x < y {
+		return x
+	}
+	return y
+}