CodeForces 625 C. K-special Tables（水~）

Description
要求构造一个n*n矩阵满足以下条件：
1.矩阵每行元素递增
2.矩阵第k列元素和最小
任意输出一种满足条件的情况
Input
两个整数n和k(1<=n<=500,1<=k<=n)
Output
输出第k列元素和的最小值并输出一个满足条件的矩阵
Sample Input
4 1
Sample Output
28
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
Solution
简单题，简单分析后可知第k列元素应为首项为n * (k-1)+1，公差为n-k+1的等差数列，故第k列元素和最小值为n * ((k-1) * n+1)+(n-1) * n/2 * (n-k+1)，至于构造矩阵，如果是前(k-1)列就从1开始输入，每次加一，如果是后n-k+1列则从n * (k-1)+1开始输入，每次加一
Code

#include<cstdio>
#include<iostream>
using namespace std;
int n,k;
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        int t1=1,t2=(k-1)*n+1;
        printf("%d\n",n*((k-1)*n+1)+(n-1)*n/2*(n-k+1));
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            {
                if(j<k)printf("%d",t1++);
                else printf("%d",t2++);
                printf("%c",j==n?'\n':' ');
            }
    }
    return 0;
}