求解磁铁分组问题：是否存在更简洁高效的内存优化解法？

问题回顾 #

给定数字n，输入n行格式为'10'或'01'的磁铁，磁铁通过异极相互连接（相邻磁铁字符串不同时可连接为一组，相同时为不同组），需返回最终的磁铁分组数g。

你的原实现代码 #

from itertools import tee

n = int(input())
magnets = []

for i in range(n):
    magnet = input()
    magnets.append(magnet)

def pairwise(iterable):
    a, b = tee(iterable)
    next(b, None)
    return zip(a, b)

m = iter(magnets)

g = 1 # there will always be at least 1 group

for x, nxt in pairwise(m):
    if x != nxt:
        g += 1
    else:
        pass

print(g)

更简便的相邻元素比较方式 #

你的代码逻辑是正确的，确实有更简洁的方式处理相邻元素对比，无需自定义pairwise函数：

1. 直接用zip拼接原列表与切片
   利用列表切片magnets[1:]获取从第二个元素开始的子列表，再和原列表magnets用zip配对，就能直接得到所有相邻元素对，同时用列表推导式简化输入收集：

   n = int(input())
   magnets = [input() for _ in range(n)]
   
   g = 1
   for prev, curr in zip(magnets, magnets[1:]):
       if prev != curr:
           g += 1
   print(g)
   

2. 用sum函数一行统计分组数
   分组数本质等于1 + 相邻元素不同的次数，可以用生成器表达式直接统计不同次数，再加上初始的1，代码更紧凑且可读性不受影响：

   n = int(input())
   magnets = [input() for _ in range(n)]
   g = 1 + sum(a != b for a, b in zip(magnets, magnets[1:]))
   print(g)
   

这两种方式都省去了自定义函数的步骤，代码更简洁但依然清晰，符合你想要的简便需求。

内容的提问来源于stack exchange，提问作者Blink