fork download
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3. int main()
  4. {
  5. ios_base::sync_with_stdio(false);
  6. cin.tie(NULL);
  7. cout.tie(NULL);
  8.  
  9. int t;
  10. cin>>t;
  11. while(t--)
  12. {
  13. int n,m,i,j,l=0;
  14. cin>>m>>n;
  15. unsigned long long int a[m][n],v;
  16. cin>>v;
  17. for(i=0;i<m;i++)
  18. {
  19. for(j=0;j<n;j++)
  20. {
  21. cin>>a[i][j];
  22. }
  23. }
  24. for(int k=0;k<m+n-1;k++)
  25. {
  26. int d;
  27. if(k<n)
  28. {
  29. d=0;
  30. }
  31. else
  32. {
  33. d=k-n+1;
  34. }
  35. if(k%2!=0)
  36. {
  37. for(j=d;j<=k-d;j++)
  38. {
  39. if(a[j][k-j]==v)
  40. { cout<<a[j][k-j];
  41. l=1;
  42. break;
  43. }
  44. else
  45. {
  46. cout<<a[j][k-j]<<", ";
  47. }
  48. }
  49. }
  50. else
  51. {
  52. for(j=k-d;j>=d;j--)
  53. {
  54. if(a[j][k-j]==v)
  55. {
  56. cout<<a[j][k-j];
  57. l=1;
  58. break;
  59. }
  60. else
  61. {
  62. cout<<a[j][k-j]<<", ";
  63. }
  64. }
  65. }
  66. if(l==1)
  67. {
  68. break;
  69. }
  70. }
  71. cout<<"\n";
  72. }
  73. }
Success #stdin #stdout 0s 4172KB
stdin
1
5 5 20
1 2 3 4 5
6 7 8 9 10
11 12 13 14 15
16 17 18 19 20
21 22 23 24 25
stdout
1, 2, 6, 11, 7, 3, 4, 8, 12, 16, 21, 17, 13, 9, 5, 10, 14, 18, 22, 23, 19, 15, 20