通过集合判断元素是否相同

一位同学发来这样一道Python题：

这个题可以用集合来做，集合具有互异性，只要每次给集合传入n个元素，判断其长度是否为1，即可判断这几个元素是否相同。

def func(lst,n):
count=0
lst2=[]
for i in range(len(lst)-n+1):
s=lst[i:i+n]
myset=set(s)
if len(myset)==1:
count+=1
lst2.append(lst[i])
print(count)
for j in range(len(lst2)):
for k in range(n-1):
print(lst2[j],end="")
print(",",end="")
print(lst2[j])

n=int(input())
str=input()
list=[]
for s in str:
if s!=’,
list.append(s)
func(list,n)

一开始是以字符串的形式输入，“，”没有什么功能，所以在判断的时候忽略掉了。最后打印的时候再添加上。
每次从列表中给入连续的3个元素到集合中，通过集合判断元素是否相等，个人感觉这个算法挺妙的。本文完，欢迎评论区探讨