python约瑟夫环拓展问题求大神解答

def YueSeFu():
    s=input("请输入n k:")
    n=int(s.split(" ")[0])
    k=int(s.split(" ")[1])
    n_list=[1 for i in range(1,n+1)]#每个人的状态，1表示还在，0不在
    j=0#要数的数
    i=0#从第一位开始数，列表下标从0开始
    sum_1=n
    if k<2 or n<k:
        print("Data Error!")
    else:
        while sum_1>k-1:
            if n_list[i]==1:#该位置还有人
                j+=1#有人就数数    
            if j==k:
                n_list[i]=0#该位置要离开
                j=0#重新数数
                sum_1-=1#人数减一
            i=(i+1)%n
    for i in range(n):
        if n_list[i]==1:
            print(i+1,end=" ")