Gửi bạn ! UNIX

#include <bits/stdc++.h>
#define ll long long
using namespace std;
vector<vector<ll>> t(int n) { // tạo mt
    vector<vector<ll>> m(n, vector<ll>(n));
    ll x = 1;
    for (ll s=0; s<2*n-1; ++s) { 
         ll r=s<n ? s : n-1; // trên xuống 3 ngôi
         ll c=s<n ? 0 : s-(n-1); // phải trái
         while (r>=0 && c<n) m[r--][c++] =x++;
    }
    return m;
}
ll tong(const vector<vector<ll>>& m, ll k, int n) { // tính tổng đc 
    for (ll i=0;i<n;++i)
        for (ll j=0; j<n; ++j)
              if (m[i][j]==k) {
                   ll s=i+j;  // tính chỉ số
                  ll t=0;
                  // số lg số trên đg chéo
                 for (ll x=0; x<n;++x)
                      for (ll y=0; y<n;++y)
                          if (x+y==s) t+=m[x][y];
                 return t; 
            }
    return 0;
}
int main(){
    int n;
    ll k;
    cin >> n>>k;
    cout <<tong(t(n),k,n)<<endl; 
}