/*
* Copyright (C) 2018 Paweł Widera
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details:
* http://www.gnu.org/licenses/gpl.html
*/
#include "message.h"
#include "teatr.h"
#define rows GetN
#define read GetElement
#define node_id MyNodeId
#define nodes NumberOfNodes
#define push PutInt
#define pop GetInt
#define push_long PutLL
#define pop_long GetLL
#define send Send
#define receive Receive
#include <vector>
#include <iostream>
using namespace std;
int merge(vector<int>::iterator begin, vector<int>::iterator middle, vector<int>::iterator end) {
int inversions = 0;
for (;begin < middle; ++begin) {
// compare elements from left and right halfs
if (*begin > *middle) {
// insert right element to the left half
int value = *begin;
*begin = *middle;
++inversions;
// insert left element to the right half
vector<int>::iterator i = middle+1;
// shift elements left if needed
while (i < end && *i < value) {
swap(*(i-1), *i);
++i;
++inversions;
}
*(i-1) = value;
}
}
return inversions;
}
int merge_sort(vector<int>::iterator begin, vector<int>::iterator end) {
int size = end - begin;
if (size < 2) {
return 0;
}
vector<int>::iterator middle = begin + size / 2;
return merge_sort(begin, middle) + merge_sort(middle, end) + merge(begin, middle, end);
}
int main() {
int n = rows();
int k = nodes();
if (node_id() > 0) {
return 0;
}
vector<int> heights;
heights.reserve(n);
for (int i = 0; i < n; ++i) {
heights.emplace_back(read(i));
}
int invertions = merge_sort(heights.begin(), heights.end());
cout << invertions << endl;
return 0;
}
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 | /* * Copyright (C) 2018 Paweł Widera * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details: * http://www.gnu.org/licenses/gpl.html */ #include "message.h" #include "teatr.h" #define rows GetN #define read GetElement #define node_id MyNodeId #define nodes NumberOfNodes #define push PutInt #define pop GetInt #define push_long PutLL #define pop_long GetLL #define send Send #define receive Receive #include <vector> #include <iostream> using namespace std; int merge(vector<int>::iterator begin, vector<int>::iterator middle, vector<int>::iterator end) { int inversions = 0; for (;begin < middle; ++begin) { // compare elements from left and right halfs if (*begin > *middle) { // insert right element to the left half int value = *begin; *begin = *middle; ++inversions; // insert left element to the right half vector<int>::iterator i = middle+1; // shift elements left if needed while (i < end && *i < value) { swap(*(i-1), *i); ++i; ++inversions; } *(i-1) = value; } } return inversions; } int merge_sort(vector<int>::iterator begin, vector<int>::iterator end) { int size = end - begin; if (size < 2) { return 0; } vector<int>::iterator middle = begin + size / 2; return merge_sort(begin, middle) + merge_sort(middle, end) + merge(begin, middle, end); } int main() { int n = rows(); int k = nodes(); if (node_id() > 0) { return 0; } vector<int> heights; heights.reserve(n); for (int i = 0; i < n; ++i) { heights.emplace_back(read(i)); } int invertions = merge_sort(heights.begin(), heights.end()); cout << invertions << endl; return 0; } |
English