题目描述
小 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\),然后再将前后的数合并,即将要删掉的数删掉。重复操作即可,细节看代码就行。我对我马蜂的美丽程度还是很自信的。
代码
cpp
#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;
}