dp 第二维应该开到 $a_n\times \max(a_n, n)$，而不是 $a_n^2$。