fork download 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. 	int t;
  6. 	cin>>t;
  7. 	while(t--){
  8. 	    int n,m;
  9. 	    cin>>n>>m;
  10. 	    vector<int> A(1, 0);
  11. 	    for(int i=1;i<=m;i++){
  12. 	        int element;
  13. 	        cin>>element;
  14. 	        A.push_back(element);
  15. 	    }
  16.  
  17. 	    vector<long long> aarsi;
  18. 	    vector<long long> krypton;
  19.  
  20. 	    for(int target=1;target<=m;target++)
  21. 	    {
  22. 	        long long aarsi_total = 0;   
  23. 	        long long krypton_total = 0;
  24.  
  25. 	        for(int i=1;i<=m;i++)
  26. 	        {
  27. 	            int moves = abs(target - i);
  28. 	            if(A[i]==1){
  29. 	                aarsi_total += moves;
  30. 	            }
  31. 	            else if(A[i]==2){
  32. 	                krypton_total += moves;
  33. 	            }
  34. 	            else if(A[i]==3){
  35. 	                aarsi_total += moves;
  36. 	                krypton_total += moves;
  37. 	            }
  38. 	        }
  39. 	        aarsi.push_back(aarsi_total);
  40. 	        krypton.push_back(krypton_total);
  41. 	    }
  42.  
  43. 	    for(int i=0; i<m; i++){
  44. 	        long long finalAnswer = abs(aarsi[i] - krypton[i]);
  45. 	        cout << finalAnswer << " ";
  46. 	    }
  47. 	    cout<<"\n";
  48. 	}
  49.     return 0;
  50. }
Success #stdin #stdout 0s 5308KB
comments (?)
stdin 
2
1 5
2 1 3 0 1
1 8
1 1 3 0 2 0 3 3
stdout 
5 2 1 0 1 
3 2 1 4 7 8 9 10
https://ideone.com/u0QClE
language:
C++ (gcc 8.3)
created:
5 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!