Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <iostream>
- //#include <cmath>
- int main(){
- int n;
- std::cin >> n;
- int a[n][n];
- for(int i = 0; i < n; ++i)
- for(int j = 0; j < n; ++j){
- std::cin >> a[i][j];
- }
- int k;
- std::cin >> k;
- if(k > 0)
- {
- for(int i = k; i < n; ++i)
- {
- for(int j = 0; j < n; ++j)
- {
- if(i == j + 1 )
- std::cout << a[i][i - k] << ' ';
- }
- }
- }
- else if(k < 0)
- {
- for(int i = 0; i < n; ++i)
- {
- for(int j = std::abs(k); j < n; ++j)
- {
- if(j == i + 1 )
- std::cout << a[j - std::abs(k)][j] << ' ';
- }
- }
- }
- else
- {
- for(int i = 0; i < n; ++i)
- {
- for(int j = 0; j < n; ++j)
- {
- if(i == j)
- std::cout << a[i][j];
- }
- }
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement