군수열

1, 1, 2, 1, 2, 3, 1, 2, 3, 4, 1, 2, 3, 4, 5, 1, 2, 3, ...

각 군이 등차수열인 군수열 n번째 항 찾기

inputNum = int(input('n 항까지 : '))

flag = True
n = 1; nCnt = 1; target = 0

while flag:

    for i in range(1, (n+1)):
        if i == n:
            print('{}'.format(i))
        else :
            print('{}, '.format(i), end ='')
        if inputNum == nCnt:
            flag = False
            target = i
            break
        nCnt += 1

    n += 1

print('\n')
print('{}번째 항:{}'.format(inputNum,target))