对角矩阵的压缩

对角矩阵的特征是：所有非0元素都集中在以主对角线为中心的3条对角线区域，其他区域的元素都为0。
三对角矩阵第一行和最后一行为两个元素，其余行元素为三
则在a[i][j]之前的元素个数为
第1行：2
第2行：3
第3行：3
...
第i-1行：3
因为对角线i=j所以当j-i=0时a[i][j]前有1个元素
所以第i行为j-i+1个元素
则k=2+3*(i-2)+j-i+1=2i+j-3