求问大佬们第一题用python怎么写啊，就是k size那个