P1311 选择客栈
开始写了一个O(n3)的算法,只得了60,后来思考(找题解),得到了一个O(nk)的算法
其实就是一种预处理的思想,对于每一个客栈而言,只要我们预处理出他前面可以匹配的客栈数量,就可以了。
所以我们记录a[i]为颜色i的数量,b[i]为颜色i可匹配的数量
输入每一个客栈,我们都更新a[i],同时如果可匹配,更新b[i],这样我们就在读入的同时预处理,在预处理的同时计算,完成了算法。
这是一个很优美的算法, 值得学习。
#include <bits/stdc++.h>
using namespace std;
const int maxn = 200005;
int main() {
int n, k, p;
cin >> n >> k >> p;
int ans = 0;
int a[maxn], b[maxn];
for(int i = 1; i <= n; i++) {
int c, v;
cin >> c >> v;
a[c]++;
if(v <= p) {
for(int j = 0; j < k; j++) {
b[j] = a[j];
}
ans += b[c]-1;
}
else {
ans += b[c];
}
}
cout << ans;
}