diff --git "a/lcp/LCP 68. \347\276\216\350\247\202\347\232\204\350\212\261\346\235\237/README.md" "b/lcp/LCP 68. \347\276\216\350\247\202\347\232\204\350\212\261\346\235\237/README.md" index dca809e537dc9..ce88be9f1561f 100644 --- "a/lcp/LCP 68. \347\276\216\350\247\202\347\232\204\350\212\261\346\235\237/README.md" +++ "b/lcp/LCP 68. \347\276\216\350\247\202\347\232\204\350\212\261\346\235\237/README.md" @@ -150,6 +150,37 @@ func beautifulBouquet(flowers []int, cnt int) (ans int) { } ``` +#### Swift + +```swift +class Solution { + func beautifulBouquet(_ flowers: [Int], _ cnt: Int) -> Int { + let mod = Int(1e9 + 7) + var maxFlower = 0 + for flower in flowers { + maxFlower = max(maxFlower, flower) + } + + var flowerCount = [Int](repeating: 0, count: maxFlower + 1) + var ans = 0 + var j = 0 + + for i in 0.. cnt { + flowerCount[flowers[j]] -= 1 + j += 1 + } + + ans = (ans + (i - j + 1)) % mod + } + + return ans + } +} +``` + diff --git "a/lcp/LCP 68. \347\276\216\350\247\202\347\232\204\350\212\261\346\235\237/Solution.swift" "b/lcp/LCP 68. \347\276\216\350\247\202\347\232\204\350\212\261\346\235\237/Solution.swift" new file mode 100644 index 0000000000000..c94b4a21c3413 --- /dev/null +++ "b/lcp/LCP 68. \347\276\216\350\247\202\347\232\204\350\212\261\346\235\237/Solution.swift" @@ -0,0 +1,26 @@ +class Solution { + func beautifulBouquet(_ flowers: [Int], _ cnt: Int) -> Int { + let mod = Int(1e9 + 7) + var maxFlower = 0 + for flower in flowers { + maxFlower = max(maxFlower, flower) + } + + var flowerCount = [Int](repeating: 0, count: maxFlower + 1) + var ans = 0 + var j = 0 + + for i in 0.. cnt { + flowerCount[flowers[j]] -= 1 + j += 1 + } + + ans = (ans + (i - j + 1)) % mod + } + + return ans + } +} \ No newline at end of file