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
stdin
1 8
1 1 3 0 2 0 3 3
stdout
3
2
1
4
7
8
9
10