https://codeforces.com/contest/1511/problem/C
题目大意
给出 $n \ (2 \le n \le 3 \times 10^5)$ 长的数组 $a$ 其中 $1 \le a_i \le 50$。给出 $q$ 个询问,每次问最左侧的值为 $v_i$(保证一定在数组中)的元素的下标,询问后将其挪到数组最左侧。
简要题解
注意到其实每个数字只有最左侧的有用,因此,这题利用 $a_i$ 的范围很小,最终只会有不超过 $50$ 个元素会变位置。这样以来除了某个数字第一次被询问,一定出现在开始的段里,我们单独维护这一段,即便是用数组维护也很快。对于首次出现,我们只需看有多少在它右面的数被询问过了,这些数字都被插到了左面,而其他本身就在它左边的数不会跑到它有边。
数据量很大的话,可以上平衡树。做到 $O(q \log n)$ 的复杂度,小确实没必要了。
复杂度
$T$:$O(n + q\max(a_i))$
$S$:$O(n + q)$
代码实现
#include <bits/stdc++.h>
using namespace std;
int io_=[](){ ios::sync_with_stdio(false); cin.tie(nullptr); return 0; }();
using LL = long long;
using ULL = unsigned long long;
using LD = long double;
using PII = pair<int, int>;
using VI = vector<int>;
using MII = map<int, int>;
template<typename T> void cmin(T &x,const T &y) { if(y<x) x=y; }
template<typename T> void cmax(T &x,const T &y) { if(x<y) x=y; }
template<typename T> bool ckmin(T &x,const T &y) {
return y<x ? (x=y, true) : false; }
template<typename T> bool ckmax(T &x,const T &y) {
return x<y ? (x=y, true) : false; }
template<typename T> void cmin(T &x,T &y,const T &z) {// x<=y<=z
if(z<x) { y=x; x=z; } else if(z<y) y=z; }
template<typename T> void cmax(T &x,T &y,const T &z) {// x>=y>=z
if(x<z) { y=x; x=z; } else if(y<z) y=z; }
// mt19937 rnd(chrono::system_clock::now().time_since_epoch().count());
// mt19937_64 rnd_64(chrono::system_clock::now().time_since_epoch().count());
/*
---------1---------2---------3---------4---------5---------6---------7---------
1234567890123456789012345678901234567890123456789012345678901234567890123456789
*/
const int M = 50;
void solve() {
int n, q; cin >> n >> q;
vector<int> a(n);
for (int& i : a) cin >> i;
vector<int> qs(q);
for (int& i : qs) cin >> i;
vector<int> fs(M + 1);
for (int i = 0; i < n; i++) {
if (!fs[a[i]]) fs[a[i]] = i + 1;
}
vector<int> inq(M + 1);
vector<int> bg;
for (int i = 0; i < q; i++) {
int v = qs[i];
if (inq[v]) {
int p = 0;
for (int j = 0; j < (int)bg.size(); j++) {
if (bg[j] == v) {
p = j;
break;
}
}
// cerr << v << ' ' << p << endl;
cout << (p + 1);
bg.erase(bg.begin() + p);
bg.insert(bg.begin(), v);
} else {
int ans = fs[v];
for (int i = 1; i <= M; i++) {
if (inq[i] && fs[i] && fs[i] > fs[v]) ans++;
}
cout << ans;
inq[v] = 1;
bg.insert(bg.begin(), v);
}
cout << (" \n"[i == q - 1]);
// for (int i : bg) cerr << i << ' ';
// cerr << endl;
}
}
int main() {
int t = 1;
// cin >> t;
while (t--) {
solve();
}
return 0;
}
Next: [CF] E. Pencils and Boxes - Educational Codeforces Round 44 (Rated for Div. 2)