题目描述
\(link\)
小 D 正在研究交换。
小 D 认为一个整数序列是好的,当且仅当它先(不严格)上升,后(不严格)下降。
形式化地,我们认为序列 \(𝑎_1,𝑎_2,...,𝑎_𝑛\) 是好的,当且仅当存在某个 \(𝑘∈[1,𝑛]\),使得对于任意
\(1 ≤𝑖 <𝑘\),有 \(𝑎_𝑖 ≤𝑎_𝑖+1\),且对于任意 \(𝑘≤𝑖<𝑛\),有 \(𝑎_𝑖 \ge 𝑎_𝑖+1\)。
小 D 得到了一个长度为 𝑛 的序列 \(𝑎_1,𝑎_2,...,𝑎_𝑛\),他想让这个序列变成好的。
小 D 每次可以交换相邻的两个元素。因为交换很累,所以小 D 想知道,他最少需要交换多少次。
这下小 D 不会了,请你帮帮他。
subtask1(15pts):\(𝑛 ≤ 10\);
subtask2(20pts):\(𝑛 ≤ 500\);
subtask3(15pts):\(𝑛 ≤ 5000\);
subtask4(15pts):\(𝑛 ≤ 10^5\);
subtask5(20pts):\({𝑎}\) 是一个 \([1,𝑛]\) 的排列。
subtask6(15pts):无特殊限制。
对于 100% 的数据,\(1≤𝑛≤3×10^5,1≤𝑎_𝑖 ≤𝑛\)。
方法
考虑贪心。
对于一个序列,如果我们要让他变好,其最小值一定会被移到最左边或最右边。移到最左边或最右边就取决于移到哪边需要的次数最少。
于是,我们再考虑分治。每次将对序列中最小的数移到最边上,给 \(ans\) 加上移动次数,并将这个数删掉。这样,我们就又得到了一个序列。对序列进行这样的重复操作,直到 \(n\) 个数全部被删除,输出 \(ans\) 即可。
但是,当我们有很多个并列最小的数时,对这些数删除的顺序是有讲究的。每次只能删除最左边或最右边的数,否则一定会产生两个相等的数交换位置的情况。这无疑是无意义的,会使答案不是最优。并且,应删除最左边或最右边中删除所需次数更少的一边。这样,才能保证后面被删除的数是最优的。
时间复杂度:\(O(n^2)\)
优化
观察到 \(n\) 最大能达到 \(3×10^5\) ,\(n ^ 2\) 是肯定卡不过去的。这时候我们便需要优化。
我们可以使用平衡树\(fhq-treap\)。其实树状数组也可以用,又快又好些,吹普常数大的没边,但我是范浩强吹普死忠粉,我就要用。 维护 \(treap\) 对一个序列分成三段,一段为要删的数前的数,一段为其自己,一段为其后面的数,启动次数就是前面的数的 \(size\) 和后面的数的 \(size\) 取 \(min\),然后再将前后的数合并,即将要删掉的数删掉。重复操作即可,细节看代码就行。我对我马蜂的美丽程度还是很自信的。
代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define usd unsigned
#define el cout << '\n'
#define lowbit(x) (x & (-x))
const int ranmod = 1e7;
#define random ((rand() * rand()) % ranmod)
#define AC return
#define AK return 0
#define YS cout << "YES"
#define NO cout << "NO"
#define Ys cout << "Yes"
#define No cout << "No"
#define ys cout << "yes"
#define no cout << "no"
#define ls(i) ch[i][0]
#define rs(i) ch[i][1]
#define debug(num) cerr << #num << ' ' << num << '\n'
// void init();
void main_();
signed main() {ios :: sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);// freopen(".in", "r", stdin);// freopen(".out", "w", stdout);int t = 1;// cin >> t;while(t--) {// init();main_();}AK;
}
const int mod = 95225987, maxn = 3 * 1e5 + 18;int ch[maxn][2], rnd[maxn], val[maxn], siz[maxn], tot, rt;
int add(int num) {tot++;ls(tot) = rs(tot) = 0;rnd[tot] = random;siz[tot] = 1;val[tot] = num;return tot;
}
void push_up(int i) {siz[i] = siz[ls(i)] + siz[rs(i)] + 1;
}
int merge(int l, int r) {if(l * r == 0) return l + r;int res = 0;if(rnd[l] < rnd[r]) {res = l;rs(l) = merge(rs(l), r);} else {res = r;ls(r) = merge(l, ls(r));}push_up(res);return res;
}
void split(int rt, int v, int &l, int &r) {if(rt == 0) {l = 0, r = 0; AC;}if(val[rt] <= v) {l = rt;split(rs(l), v, rs(l), r);push_up(l);} else {r = rt;split(ls(r), v, l, ls(r));push_up(r);}
}
int show(int id) {int a, b, mid;split(rt, id, a, b);split(a, id - 1, a, mid);int res = min(siz[a], siz[b]);debug(siz[a]), debug(siz[b]);rt = merge(a, merge(mid, b));return res;
}
int del(int id) {int a, b, mid;split(rt, id, a, b);split(a, id - 1, a, mid);int res = min(siz[a], siz[b]);debug(siz[a]), debug(siz[b]);rt = merge(a, b);return res;
}int n, ans = 0;
map <int, deque <int> > a;void main_() {cin >> n;for(int i = 1; i <= n; i++) {int num;cin >> num;rt = merge(rt, add(i));a[num].push_back(i);}for(auto tmp : a) {while(!tmp.second.empty()) {if(tmp.second.size() == 1) {ans += del(tmp.second.front());tmp.second.pop_front();} else {int l = show(tmp.second.front()), r = show(tmp.second.back());if(l <= r) {ans += del(tmp.second.front());tmp.second.pop_front();} else {ans += del(tmp.second.back());tmp.second.pop_back();}}}}cout << ans; el;
}