Description
箱子再分配问题需要解决如下问题:
(1)一共有N个物品,堆成M堆。
(2)所有物品都是一样的,但是它们有不同的优先级。
(3)你只能够移动某堆中位于顶端的物品。
(4)你可以把任意一堆中位于顶端的物品移动到其它某堆的顶端。若此物品是当前所有物品中优先级最高的,可以直接将之删除而不用移动。
(5)求出将所有物品删除所需的最小步数。删除操作不计入步数之中。
(6)只是一个比较难解决的问题,这里你只需要解决一个比较简单的版本:
不会有两个物品有着相同的优先级,且M=2
Input
第一行是包含两个整数N1,N2分别表示两堆物品的个数。
接下来有N1行整数按照从顶到底的顺序分别给出了第一堆物品中的优先级,数字越大,优先级越高。
再接下来的N2行按照同样的格式给出了第二堆物品的优先级。
Output
对于每个数据,请输出一个整数,即最小移动步数。
Sample Input
3 3
1
4
5
2
7
3
1
4
5
2
7
3
Sample Output
6
HINT
1<=N1+N2<=100000
其实是想出来正解了,但是写搓了2333...
就是把两堆怼一块,然后树状数组那位置加一。
然后从大到小到x,找第一个比它大的y,用然后把两个地方求和做差加进答案。
然后再把x位置上-1.
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #include <cmath> using namespace std; #define ll long long inline int read(){ #define g getchar() #define is isdigit(ch) int res=0;char ch=g; while(!is) ch=g; while(is){res=(res<<3)+(res<<1)+(ch^48);ch=g;} #undef d #undef is return res; } int n1, n2, n; struct date{ int val, p; }a[100005]; int maxx, p; ll ans; inline bool cmp(date a, date b){return a.val > b.val;} int t[100005*20]; #define lowbit x & -x inline void add(int x, int d) { while(x <= n) { t[x] += d;x += lowbit; } } inline int ask(int x) { int res = 0; while(x) { res += t[x]; x -= lowbit; } return res; } inline int query(int x, int y) { if (x > y) swap(x, y); return ask(y) - ask(x - 1); } int main() { n1 = read(), n2 = read(); n = n1 + n2 + 1; for (int i = 1 ; i <= n1 ; i ++) { a[n1-i+1].val = read(); if (a[i].val > maxx) maxx = a[i].val, p = 1; } int now = n1 + 1; for (int i = 1 ; i <= n2 ; i ++) { a[i+n1+1].val = read(); if (a[i+n1+1].val > maxx) maxx = a[i+n1+1].val, p = 2; } for (int i = 1 ; i <= n ; i ++) { if (i != now) add(i, 1); a[i].p = i; } sort(a + 1, a + 1 + n, cmp); for (int i = 1 ; i < n ; i ++) { ans += query(a[i].p, now) - 1; now = a[i].p; add(now, -1); } printf("%lld ", ans); return 0; }