A^2=0 等价于 A 的所有特征值为 0 且 Jordan 块不超过 2 阶
从上述矩阵中剔除零矩阵就是你需要的