पायथन में कॉनेल सीक्वेंस

समस्या विवरण

कॉनेल सीक्वेंस का nth टर्म ज्ञात करें। सीक्वेंस इस प्रकार बनता है:

1. पहला विषम पूर्णांक: 1
2. अगले दो सम पूर्णांक: 2, 4
3. अगले तीन विषम पूर्णांक: 5, 7, 9
4. अगले चार सम पूर्णांक: 10, 12, 14, 16
5. और इसी तरह...

उदाहरण

इनपुट: 12

आउटपुट: 21

एल्गोरिदम

1. i = 1 से शुरू करें
2. जब तक i*(i+1)/2 < n+1, i++ करें
3. idx = i*(i+1)//2
4. num = i²
5. वापस करें: num - 2*(idx - n - 1)

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

class Solution:
    def solve(self, n):
        i = 1
        while (i * (i + 1) // 2) < n + 1:
            i += 1
        idx = i * (i + 1) // 2
        num = i ** 2
        return num - 2 * (idx - n - 1)

ob = Solution()
print(ob.solve(12))

इनपुट

12

आउटपुट

21