2141: 排队
Time Limit: 4 Sec Memory Limit: 259 MB
Description
排排坐,吃果果,生果甜嗦嗦,大家笑呵呵。你一个,我一个,大的分给你,小的留给我,吃完果果唱支歌,大家乐和和。红星幼儿园的小朋友们排起了长长地队伍,准备吃果果。不过因为小朋友们的身高有所区别,排成的队伍高低错乱,极不美观。设第i个小朋友的身高为hi,我们定义一个序列的杂乱程度为:满足ihj的(i,j)数量。幼儿园阿姨每次会选出两个小朋友,交换他们的位置,请你帮忙计算出每次交换后,序列的杂乱程度。为方便幼儿园阿姨统计,在未进行任何交换操作时,你也应该输出该序列的杂乱程度。
Input
第一行为一个正整数n,表示小朋友的数量;第二行包含n个由空格分隔的正整数h1,h2,…,hn,依次表示初始队列中小朋友的身高;第三行为一个正整数m,表示交换操作的次数;以下m行每行包含两个正整数ai和bi¬,表示交换位置ai与位置bi的小朋友。
Output
输出文件共m行,第i行一个正整数表示交换操作i结束后,序列的杂乱程度。
Sample Input
3
130 150 140
2
2 3
1 3
Sample Output
1
0
3
HINT
【样例说明】
未进行任何操作时,(2,3)满足条件;
操作1结束后,序列为130 140 150,不存在满足ihj的(i,j)对;
操作2结束后,序列为150 140 130,(1,2),(1,3),(2,3)共3对满足条件的(i,j)。
【数据规模和约定】
对于100%的数据,1≤m≤2103,1≤n≤2104,1≤hi≤109,ai≠bi,1≤ai,bi≤n。
题解
此题就是求一个动态的逆序对个数,而且修改操作只有交换,所以是比较简单的。
我们采用分块的方法,将数组没\(150\)分一块,每次操作我们只用暴力枚举分块内的和每个分块。
复杂度约为\(O(nlogn \sqrt n)\)。
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 20005
#define SIZE 150
int ina; char inc, inb[1<<16], *ins = inb, *ine = inb;
#define getc() ((ins==ine&&(ine=(ins=inb)+fread(inb,1,1<<16,stdin),ins==ine))?EOF:*ins++)
inline int geti() {
while ((inc = getc()) < '0' || inc > '9'); ina = inc - '0';
while ((inc = getc()) >= '0' && inc <= '9') ina = (ina << 3) + (ina << 1) + inc - '0';
return ina;
}
int pre[N], C[140][N], h[N], r[N];
void Add(int *tre, int x, int val) {
for (; x < N; x += x & -x)
tre[x] += val;
}
int Sum(int *tre, int x) {
int ret = 0;
for (; x > 0; x -= x & -x)
ret += tre[x];
return ret;
}
bool cmp(const int &a, const int &b) { return h[a] < h[b]; }
int main() {
int n, i, j, ans = 0, m, id, la, x, y, idx, idy;
for (n = geti(), i = 1; i <= n; ++i) h[i] = geti(), r[i] = i;
std::sort(r + 1, r + n + 1, cmp);
la = h[r[1]]; h[r[1]] = id = 1;
for (i = 2; i <= n; ++i) {
if (h[r[i]] ^ la) la = h[r[i]], ++id;
h[r[i]] = id;
}
for (i = n; i; --i)
ans += Sum(pre, h[i] - 1), Add(pre, h[i], 1);
for (i = 1; i <= n; ++i) Add(C[(i-1)/SIZE], h[i], 1);
printf("%d\n", ans);
for (m = geti(); m; --m) {
x = geti(), y = geti();
if (x > y) x ^= y ^= x ^= y;
idx = (x-1) / SIZE + 1, idy = (y-1) / SIZE - 1;
if (idx <= idy) {
for (i = idx; i <= idy; ++i) {
ans -= Sum(C[i], h[x]-1);
ans += Sum(C[i], n) - Sum(C[i], h[x]);
ans += Sum(C[i], h[y]-1);
ans -= Sum(C[i], n) - Sum(C[i], h[y]);
}
for (i = x + 1, j = idx * SIZE; i <= j; ++i) {
ans -= h[i] < h[x];
ans += h[i] > h[x];
ans += h[i] < h[y];
ans -= h[i] > h[y];
}
for (i = (idy+1)*SIZE+1; i < y; ++i) {
ans -= h[i] < h[x];
ans += h[i] > h[x];
ans += h[i] < h[y];
ans -= h[i] > h[y];
}
} else {
for (i = x + 1; i < y; ++i) {
ans -= h[i] < h[x];
ans += h[i] > h[x];
ans += h[i] < h[y];
ans -= h[i] > h[y];
}
}
if (h[x] < h[y]) ++ans;
else if (h[x] > h[y]) --ans;
printf("%d\n", ans);
Add(C[(x-1)/SIZE], h[x], -1); Add(C[(y-1)/SIZE], h[y], -1);
h[x] ^= h[y] ^= h[x] ^= h[y];
Add(C[(x-1)/SIZE], h[x], 1); Add(C[(y-1)/SIZE], h[y], 1);
}
return 0;
}