[蓝桥杯 2025 省 C/Java A/研究生组] 冷热数据队列题解_若 p 不在队列 q 中(即既不在 q 1 中,也不在 q 2 中),则加载数据页 p,并插入到 q-CSDN博客

本文链接：https://blog.csdn.net/Gen_Jack/article/details/147701199

P12166 [蓝桥杯 2025 省 C/Java A/研究生组] 冷热数据队列

题目描述

冷热数据队列 $q$ 可以看做由两个子队列组成：长度为 $n_1$ 的热数据队列 $q_1$ 和长度为 $n_2$ 的冷数据队列 $q_2$ 。当我们需要访问某个数据页 $p$ 时：

若 $p$ 不在队列 $q$ 中（即既不在 $q_1$ 中，也不在 $q_2$ 中），则加载数据页 $p$ ，并插入到 $q_2$ 的首部。
若 $p$ 已经在队列 $q$ 中，则将 $p$ 移动至 $q_1$ 首部。
当 $q_1$ 或 $q_2$ 队列容量不足时，会将其尾部的数据页淘汰出去。
当 $q_1$ 已满，但 $q_2$ 未满时，从 $q_1$ 中淘汰出的数据页会移动到 $q_2$ 首部。

输入格式

输入的第一行包含两个正整数 $n_1, n_2$ ，用一个空格分隔。

第二行包含一个整数 $m$ ，表示操作次数。

第三行包含 $m$ 个正整数 $,vmv_1, v_2, \cdots, v_m$ ，表示依次访问到的数据页的编号，相邻整数之间使用一个空格分隔。

输出格式

输出两行。

第一行包含若干个整数，相邻整数之间使用一个空格分隔，依次表示 $q_1$ 中的数据页。

第二行包含若干个整数，相邻整数之间使用一个空格分隔，依次表示 $q_2$ 中的数据页。

输入输出样例 #1

输入 #1

3 3
10
1 2 3 4 3 2 2 1 3 4

输出 #1

4 3 2
1

说明/提示

样例说明

$i$	$v_i$	$q_1$	$q_2$
$-$	$-$	$[]$	$[]$
$1$	$1$	$[]$	$[1]$
$2$	$2$	$[]$	$[2, 1]$
$3$	$3$	$[]$	$[3, 2, 1]$
$4$	$4$	$[]$	$[4, 3, 2]$
$5$	$3$	$[3]$	$[4, 2]$
$6$	$2$	$[2, 3]$	$[4]$
$7$	$2$	$[2, 3]$	$[4]$
$8$	$1$	$[2, 3]$	$[1, 4]$
$9$	$3$	$[3, 2]$	$[1, 4]$
$10$	$4$	$[4, 3, 2]$	$[1]$

评测用例规模与约定

对于 $20%20\%$ 的评测用例， $\leq n_1, n_2 \leq 10$ ， $\leq m \leq 10$ ；
对于 $40%40\%$ 的评测用例， $\leq n_1, n_2 \leq 20$ ， $\leq m \leq 100$ ；
对于 $60%60\%$ 的评测用例， $\leq n_1, n_2 \leq 100$ ， $\leq m \leq 1000$ ；
对于 $80%80\%$ 的评测用例， $\leq n_1, n_2 \leq 10^3$ ， $\leq m \leq 10^4$ ；
对于所有评测用例， $\leq n_1, n_2 \leq 10^4$ ， $\leq m \leq 10^5$ ， $\leq v_i \leq 10^4$ 。

题解

这道题比较简单，就是简单地模拟，但是在实现上需要注意细节。
直接上代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e4 + 7;

struct A {
    int status = 0;
    int pre = -1, ne = -1;
}a[N];
int n1, n2, m, v;
int q1 = -1, q2 = -1, nq1 = 0, nq2 = 0, tail1 = -1, tail2 = -1;

void eraseQ2(int x) {
    if (tail2 == x)
        tail2 = a[tail2].pre;
    if (q2 == x)
        q2 = a[x].ne;
    else {
        a[a[x].pre].ne = a[x].ne;
        if (a[x].ne != -1)
            a[a[x].ne].pre = a[x].pre;
    }
    a[x].ne = -1;
    a[x].pre = -1;
    a[x].status = 0;
    nq2--;
}
void popQ2() {
    int tmp = tail2;
    tail2 = a[tail2].pre;
    a[tail2].ne = -1;
    a[tmp].status = 0;
    a[tmp].pre = -1;
    a[tmp].ne = -1;
    nq2--;
}
void pushQ2(int x) {
    a[x].status = 2;
    if (q2 == -1)
        q2 = x, tail2 = x;
    else {
        a[x].ne = q2;
        a[q2].pre = x;
        q2 = x;
    }
    nq2++;
    if (nq2 > n2)
        popQ2();
}

void eraseQ1(int x) {
    if (tail1 == x)
        tail1 = a[tail1].pre;
    if (q1 == x)
        q1 = a[x].ne;
    else {
        a[a[x].pre].ne = a[x].ne;
        if (a[x].ne != -1)
            a[a[x].ne].pre = a[x].pre;
    }
    a[x].ne = -1;
    a[x].pre = -1;
    a[x].status = 0;
    nq1--;
}
void popQ1() {
    int tmp = tail1;
    tail1 = a[tail1].pre;
    a[tail1].ne = -1;
    a[tmp].status = 0;
    a[tmp].pre = -1;
    a[tmp].ne = -1;
    nq1--;
    pushQ2(tmp);
}
void pushQ1(int x) {
    a[x].status = 1;
    if (q1 == -1)
        q1 = x, tail1 = x;
    else {
        a[x].ne = q1;
        a[q1].pre = x;
        q1 = x;
    }
    nq1++;
    if (nq1 > n1)
        popQ1();
}

int main() {
    cin >> n1 >> n2;
    cin >> m;
    while (m--) {
        cin >> v;
        if (a[v].status == 0)
            pushQ2(v);
        else if (a[v].status == 2) {
            eraseQ2(v);
            pushQ1(v);
        }
        else if (a[v].status == 1) {
            eraseQ1(v);
            pushQ1(v);
        }
    }
    int p = q1;
    while (p != -1)
    {
        cout << p << " ";
        p = a[p].ne;
    }
    cout << endl;
    p = q2;
    while (p != -1)
    {
        cout << p << " ";
        p = a[p].ne;
    }
    return 0;
}