Guest User

Untitled

a guest
May 25th, 2018
90
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.15 KB | None | 0 0
  1. #include<cstdio>
  2. //#include<set>
  3. #include<vector>
  4. #include<queue>
  5. #define MAXN 200000
  6.  
  7. inline long long int min(long long int a, long long int b) { return a < b ? a : b; }
  8. inline int max(int a, int b) { return a < b ? b : a; }
  9.  
  10. struct Tree
  11. {
  12.   bool isLeaf;
  13.   int leftChild;
  14.   int rightChild;
  15.   int value;
  16.   //int deep;
  17.   int numberOfLeafs;
  18. };
  19.  
  20. struct V
  21. {
  22.   V(int a);
  23.   V(V a, V b);
  24.  
  25.   std::vector<int> numbers;
  26.   long long int inversions;
  27. };
  28.  
  29. V::V(int a)
  30. {
  31.   inversions = 0;
  32.   numbers.push_back(a);
  33. }
  34.  
  35. V::V(V a, V b)
  36. {
  37.   long long int inversionsInTheFirstCase = 0;
  38.   long long int inversionsInTheSecondCase = 0;
  39.  
  40.   //
  41.   // first case
  42.  
  43.   {
  44.     std::queue<int> q;
  45.     for(std::vector<int>::iterator i = a.numbers.begin(); i != a.numbers.end(); i++)
  46.       q.push(*i);
  47.  
  48.     for(std::vector<int>::iterator i = b.numbers.begin(); i != b.numbers.end(); i++)
  49.     {
  50.       while(!q.empty() && q.front() < (*i))
  51.         q.pop();
  52.       inversionsInTheFirstCase += q.size();
  53.     }
  54.   }
  55.  
  56.   //
  57.   // second case
  58.  
  59.   {
  60.     std::queue<int> q;
  61.     for(std::vector<int>::iterator i = b.numbers.begin(); i != b.numbers.end(); i++)
  62.       q.push(*i);
  63.  
  64.     for(std::vector<int>::iterator i = a.numbers.begin(); i != a.numbers.end(); i++)
  65.     {
  66.       while(!q.empty() && q.front() < (*i))
  67.         q.pop();
  68.       inversionsInTheSecondCase += q.size();
  69.     }
  70.   }
  71.  
  72.   //
  73.   // updating inversions number
  74.  
  75.   inversions = a.inversions + b.inversions + min(inversionsInTheFirstCase, inversionsInTheSecondCase);
  76.  
  77.   //
  78.   // merging sets
  79.  
  80.   {
  81.     int iA = 0;
  82.     int iB = 0;
  83.  
  84.     while(iA < a.numbers.size() && iB < b.numbers.size())
  85.     {
  86.       if(a.numbers[iA] < b.numbers[iB])
  87.       {
  88.         numbers.push_back(a.numbers[iA]);
  89.         iA++;
  90.       }
  91.       else
  92.       {
  93.         numbers.push_back(b.numbers[iB]);
  94.         iB++;
  95.       }
  96.     }
  97.  
  98.     while(iA < a.numbers.size())
  99.     {
  100.       numbers.push_back(a.numbers[iA]);
  101.       iA++;
  102.     }
  103.  
  104.     while(iB < b.numbers.size())
  105.     {
  106.       numbers.push_back(b.numbers[iB]);
  107.       iB++;
  108.     }
  109.   }
  110. };
  111.  
  112. int n, p, treeSize;
  113. Tree tree[3 * MAXN];
  114.  
  115. int GetInput()
  116. {
  117.   int position = treeSize;
  118.   treeSize++;
  119.  
  120.   scanf("%d", &p);
  121.   if(p != 0)
  122.   {
  123.     tree[position].isLeaf = true;
  124.     tree[position].value = p;
  125.     tree[position].numberOfLeafs = 1;
  126.   }
  127.   else
  128.   {
  129.     tree[position].isLeaf = false;
  130.     tree[position].leftChild = GetInput();
  131.     tree[position].rightChild = GetInput();
  132.     tree[position].numberOfLeafs = max(tree[tree[position].leftChild].numberOfLeafs, tree[tree[position].rightChild].numberOfLeafs);
  133.   }
  134.  
  135.   return position;
  136. }
  137.  
  138. V MakeOutput(int position)
  139. {
  140.   if(tree[position].isLeaf == true)
  141.     return V(tree[position].value);
  142.   else
  143.   {
  144.     if(tree[tree[position].leftChild].numberOfLeafs > tree[tree[position].rightChild].numberOfLeafs)
  145.       return V(MakeOutput(tree[position].leftChild), MakeOutput(tree[position].rightChild));
  146.     else
  147.       return V(MakeOutput(tree[position].rightChild), MakeOutput(tree[position].leftChild));
  148.   }
  149. }
  150.  
  151. int main()
  152. {
  153.   scanf("%d", &n);
  154.   GetInput();
  155.   printf("%lld", MakeOutput(0).inversions);
  156.   return 0;
  157. }
Add Comment
Please, Sign In to add comment