Advertisement
Korotkodul

N4

Dec 17th, 2022 (edited)
485
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.22 KB | None | 0 0
  1. #include <iostream>
  2. #include <cmath>
  3. #include <vector>
  4. #include <queue>
  5. #include <algorithm>
  6. #include <string>
  7. #include <stack>
  8. #include <set>
  9. #include <map>
  10. #define pii pair <int, int>
  11. #define pb(x) push_back(x)
  12. using namespace std;
  13. using ll = long long;
  14. using ld = long double;
  15. using db = double;
  16. void cv(vector <int> &v) {
  17.     for (auto x : v) cout << x << ' ';
  18.     cout << "\n";
  19. }
  20.  
  21. void cvl(vector <ll> &v) {
  22.     for (auto x : v) cout << x << ' ';
  23.     cout << "\n";
  24. }
  25.  
  26.  
  27. void cvv(vector <vector <int> > &v) {
  28.     for (auto x : v) cv(x);
  29.     cout << "\n";
  30. }
  31.  
  32. void cvb(vector <bool> v) {
  33.     for (bool x : v) cout << x << ' ';
  34.     cout << "\n";
  35. }
  36.  
  37. void cvs(vector <string>  v) {
  38.     for (auto a : v) {
  39.         cout << a << "\n";
  40.     }
  41. }
  42.  
  43. void cvp(vector <pii> a) {
  44.     for (auto p : a) {
  45.         cout << p.first << ' ' << p.second << "\n";
  46.     }
  47.     cout << "\n";
  48. }
  49.  
  50. vector <int> dx, dy;
  51. int n;
  52. vector <vector <int> > m, id;
  53.  
  54. bool ok1(int i, int j) {
  55.     return i >= 0 && j >= 0 && i + j < n;
  56. }
  57.  
  58. bool ok2(int i, int j) {
  59.     return i < n && j < n && i + j > n;
  60. }
  61.  
  62. bool sh = 1;
  63.  
  64. int main() {
  65.     ios::sync_with_stdio(0);
  66.     cin.tie(0);
  67.     cout.tie(0);
  68.     cin >> n;
  69.     dx = { 1, -1, 0, 1,};
  70.     dy = { 0, 1, 1, -1,};
  71.     int i = 0, j = 0;
  72.     m.assign(n, vector <int> (n, 0));
  73.     id.assign(n, vector <int> (n, 0));
  74.     int cnt = 1;
  75.     id[0][0] = 1;
  76.     int tk = 1;
  77.     while (tk != (n - 1) * n / 2) {
  78.         if (ok1(i + dx[0], j + dy[0])) {
  79.             cnt+=2;
  80.             tk++;
  81.             i = i + dx[0];
  82.             j = j + dy[0];
  83.             id[i][j] = cnt;
  84.         }
  85.         while (ok1(i + dx[1], j + dy[1])) {
  86.             cnt+=2;
  87.             tk++;
  88.             i += dx[1];
  89.             j += dy[1];
  90.             id[i][j] = cnt;
  91.         }
  92.         if (ok1(i + dx[2], j + dy[2])) {
  93.             cnt+=2;
  94.             tk++;
  95.             i += dx[2];
  96.             j += dy[2];
  97.             id[i][j] = cnt;
  98.         }
  99.         while (ok1(i + dx[3], j + dy[3])) {
  100.             cnt+=2;
  101.             tk++;
  102.             i += dx[3];
  103.             j += dy[3];
  104.             id[i][j] = cnt;
  105.         }
  106.     }
  107.  
  108.     dx = {0, -1, -1, 1};
  109.     dy = {-1, 1, 0, -1};
  110.     cnt = 2;
  111.     tk++;
  112.     id[n-1][n-1] = 2;
  113.     i = n - 1;
  114.     j = n - 1;
  115.     while (tk != (n - 1) * n) {
  116.         if (ok2(i + dx[0], j + dy[0])) {
  117.             cnt+=2;
  118.             tk++;
  119.             i = i + dx[0];
  120.             j = j + dy[0];
  121.             id[i][j] = cnt;
  122.         }
  123.         while (ok2(i + dx[1], j + dy[1])) {
  124.             cnt+=2;
  125.             tk++;
  126.             i += dx[1];
  127.             j += dy[1];
  128.             id[i][j] = cnt;
  129.         }
  130.         if (ok2(i + dx[2], j + dy[2])) {
  131.             cnt+=2;
  132.             tk++;
  133.             i += dx[2];
  134.             j += dy[2];
  135.             id[i][j] = cnt;
  136.         }
  137.         while (ok2(i + dx[3], j + dy[3])) {
  138.             cnt+=2;
  139.             tk++;
  140.             i += dx[3];
  141.             j += dy[3];
  142.             id[i][j] = cnt;
  143.         }
  144.     }
  145.     cnt++;
  146.     tk++;
  147.     i = 0;
  148.     j = n - 1;
  149.     id[i][j] = cnt;
  150.     while (tk != n * n) {
  151.         cnt++;
  152.         tk++;
  153.         i++;
  154.         j--;
  155.         id[i][j] = cnt;
  156.     }
  157.     if (sh) {
  158.         cvv(id);
  159.     }
  160. }
  161.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement