fork download 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. void scoreDifference(vector<int> &arr, int n){
  5. 	vector<int> a(n);
  6. 	vector<int> b(n);
  7. 	for (int i = 0; i < n; i++){
  8. 		if(arr[i] == 1){
  9. 			a[i] = 1;
  10. 		}
  11. 		else if(arr[i] == 2){
  12. 			b[i] = 1;
  13. 		}
  14. 		else if(arr[i] == 3){
  15. 			a[i] = 1;
  16. 			b[i] = 1;
  17. 		}
  18. 	}
  19.  
  20. 	vector<int> arsi(n);
  21. 	vector<int> krypto(n);
  22. 	for (int i = 0; i < n; i++){
  23. 		int d = 0;
  24. 		for (int j = 0; j < n; j++){
  25. 			if(a[j] == 1){
  26. 				d += abs(j - i);
  27. 			}
  28.  
  29. 		}
  30. 		arsi[i] = d;
  31. 	}
  32. 	for (int i = 0; i < n; i++){
  33. 		int d = 0;
  34. 		for (int j = 0; j < n; j++){
  35. 			if(b[j] == 1){
  36. 				d += abs(j - i);
  37. 			}
  38.  
  39. 		}
  40. 		krypto[i] = d;
  41. 	}
  42.  
  43. 	for (int i = 0; i < n; i++){
  44. 		cout << abs(arsi[i] - krypto[i]) << endl;
  45. 	}
  46. }
  47.  
  48. int main() {
  49. 	// your code goes here
  50. 	int x, n; cin >> x >> n;
  51. 	vector<int> arr(n);
  52. 	for (int &i : arr){
  53. 		cin >> i;
  54. 	}
  55.  
  56. 	scoreDifference(arr, n);
  57.  
  58. 	return 0;
  59. }
Success #stdin #stdout 0s 5320KB
comments (?)
stdin 
1 8
1 1 3 0 2 0 3 3
stdout 
3
2
1
4
7
8
9
10
https://ideone.com/Ym5Nrl
language:
C++14 (gcc 8.3)
created:
10 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!