代码：

#include <iostream>  
#include <vector>  
using namespace std;  
int main() {  
    int n, k;  
    cin >> n >> k;  
    vector<vector<int> > grid(n, vector<int>(n, 0));  
    int count = 1;
    for (int i = 0; i < n; ++i) {  
        for (int j = 0; j < n; ++j) {  
            if (grid[i][j] == 0 && (j + 1 < n && grid[i][j + 1] == 0) && !(i == 0 && j < k)) {  
                grid[i][j] = count;  
                grid[i][j + 1] = count;  
                ++count;   
                if (i + 1 < n && j + 1 == k) {  
                    j = k; 
                }  
            }  
        }  
    }  
    for (int i = 0; i < n; ++i) {  
        for (int j = 0; j < n; ++j) {  
            cout << grid[i][j] << " ";  
        }  
        cout << endl;  
    }  
    return 0;  
}

A了3、4两个点