Advertisement
Guest User

Untitled

a guest
Aug 23rd, 2017
53
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.86 KB | None | 0 0
  1. #include <iostream>
  2. //#include <cmath>
  3. int main(){
  4. int n;
  5. std::cin >> n;
  6. int a[n][n];
  7. for(int i = 0; i < n; ++i)
  8. for(int j = 0; j < n; ++j){
  9. std::cin >> a[i][j];
  10. }
  11. int k;
  12. std::cin >> k;
  13. if(k > 0)
  14. {
  15. for(int i = k; i < n; ++i)
  16. {
  17. for(int j = 0; j < n; ++j)
  18. {
  19. if(i == j + 1 )
  20. std::cout << a[i][i - k] << ' ';
  21.  
  22.  
  23. }
  24. }
  25. }
  26. else if(k < 0)
  27. {
  28. for(int i = 0; i < n; ++i)
  29. {
  30. for(int j = std::abs(k); j < n; ++j)
  31. {
  32. if(j == i + 1 )
  33. std::cout << a[j - std::abs(k)][j] << ' ';
  34.  
  35.  
  36. }
  37. }
  38. }
  39. else
  40. {
  41. for(int i = 0; i < n; ++i)
  42. {
  43. for(int j = 0; j < n; ++j)
  44. {
  45. if(i == j)
  46. std::cout << a[i][j];
  47. }
  48. }
  49. }
  50.  
  51. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement