Codeforces Global Round 16-D2
题目大意:
给n * m个座位,有 n* m个观众,每个观众有一个视力值a[i],视力值越小的观众坐的座位编号越小,现在要去所有观众按给定的视力数组的顺序入座,每个观众在到达自己位置时要经过的其他观众的最小值。
树状数组 + 离散化 + 思维
分析
1.因为视力小的位置必须小,所以最终所有视力的位置都是唯一确定的
2.视力相同的观众的位置不固定,为了使入座经过的人数最小,对于相同视力的观众,入座顺序靠前的选择行数小,列数靠右的位置,这样的贪心策略可以保证相同视力的观众为答案提供贡献,同时保证当前顺序在后面不同视力的人(可能座位在当前位置前面)不会再增加贡献
3.经过人数的计算:当前观众入座要经过多少人其实就是当前时间点上有多少与位置与当前位置构成正序对,计算正序对的方法参考树状数组 + 离散化。
AC代码
#include <bits/stdc++.h>
#define vi vector<int>
#define pb push_back
#define pii pair<int, int>
using namespace std;
const int N = 310 * 310;
int a[N], p[N], n, m, t;
map<pii, int> tr;
vi alls;
bool cmp(int x, int y) {
if(a[x] == a[y]) return x < y;
else return a[x] < a[y];
}
void add(int r, int x, int v) {
for (int i = x; i <= n * m; i += i & -i) tr[{r, i}] += v;
}
int sum(int r, int x) {
int res = 0;
for (int i = x; i; i -= i & -i) res += tr[{r, i}];
return res;
}
void solve() {
cin >> n >> m;
for (int i = 1; i <= n * m; i ++ ) {
cin >> a[i]; alls.pb(a[i]); p[i] = i;
}
sort(alls.begin(), alls.end());
alls.erase(unique(alls.begin(), alls.end()), alls.end());
sort(p + 1, p + 1 + n * m, cmp);
map<int, int> r, c;
for (int i = 1; i <= n * m; i ++ ) {
int id = p[i];
r[id] = (i + m - 1) / m;
}
int ans = 0;
for (int i = 1; i <= n * m; i ++ ) {
ans += sum(r[i], lower_bound(alls.begin(), alls.end(), a[i]) - alls.begin());
add(r[i], lower_bound(alls.begin(), alls.end(), a[i]) - alls.begin() + 1, 1);
}
cout << ans << endl;
}
int main() {
cin >> t;
while(t -- ) {
tr.clear(); alls.clear();
solve();
}
return 0;
}