पायथन में लिस्ट में x और x+1 दोनों मौजूद होने वाले तत्वों की गिनती

समस्या विवरण

संख्याओं की एक सूची nums दी गई है, हमें ऐसे तत्वों x की संख्या गिननी है जिनके लिए x + 1 भी सूची में मौजूद है।

उदाहरण

इनपुट: [2, 3, 3, 4, 8]

आउटपुट: 3

व्याख्या: 2 (क्योंकि 3 मौजूद है), 3 (क्योंकि 4 मौजूद है), 3 (फिर से) — कुल 3

एल्गोरिदम

1. nums से एक सेट s बनाएं (O(1) लुकअप के लिए)
2. count = 0 इनिशियलाइज़ करें
3. nums के प्रत्येक तत्व i के लिए:
   • यदि i + 1 सेट s में है, तो count++
4. count वापस करें

पायथन कार्यान्वयन

class Solution:
    def solve(self, nums):
        s = set(nums)
        count = 0
        for i in nums:
            if i + 1 in s:
                count += 1
        return count

ob = Solution()
nums = [2, 3, 3, 4, 8]
print(ob.solve(nums))

इनपुट

[2, 3, 3, 4, 8]

आउटपुट

3