2019牛客暑期多校训练营（第三场） J LRU management 模拟链表操作

输入n, m，n表示n种操作，m表示最多可以容纳m个串。

第一种操作：先在容器里找是否存在这个串，如果不存在，则添加在末尾，这个串携带了一个值v。

　　　　　　如果存在，则先把之前存在的那个拿出来，然后在后面添加这个串，注意此时输入的这个v值不管用！！！！，v还是以前的那个v.

第二种操作：先在这个容器内是否存在这个串，如果存在则看v值，v值为-1， 0， 1，分别表示左边那个，自己，右边那个，如果存在则输出那个串携带的v值，不存在则输出invalid.

#include<bits/stdc++.h>
 
using namespace std;
const int M = 5e5 + ;
typedef pair<int, int> pii;
//flag表示v值，ind表示是第几个串，如果先出去了再进来算另外一个
int flag[M * ], ind[M * ];
int t, n, m, tot;
char s[];
int tree[M * ][];
//判断这个串是否出现过与是否已经出去了。下标对应cnt
int vis[M];
//左边，右边。
int le[M], re[M];
//保存v值。下标对应cnt
int ans[M];
容器头元素，尾元素。
int mi, ma;
 
void add(char ch[], int in, int val) {
    int root = , len = strlen(ch);
    for(int i = ; i < len; i++) {
        int id = ch[i] - '';
        if(!tree[root][id]) {
            memset(tree[tot], , sizeof(tree[tot]));
            ind[tot] = -;
            flag[tot] = -;
            tree[root][id] = tot++;
        }
        root = tree[root][id];
    }
    ind[root] = in;
    flag[root] = val;
}
 
pii findx(char ch[]) {
    int root = , len = strlen(ch);
    for(int i = ; i < len; i++) {
        int id = ch[i] - '';
        root = tree[root][id];
        if(!root) return pii(-, -);
    }
    return pii(ind[root], flag[root]);
}
 
int main() {
    scanf("%d", &t);
    while(t--) {
        scanf("%d%d", &n, &m);
        tot = ;
        memset(tree[], , sizeof(tree[]));
        int opt, v;
　　　　//给串标号，如果出去了再进来也要++.
        int cnt = ;
　　　　//表示容器里有几个元素。
        int num = ;
        mi = , ma = -;
        for(int i = ; i <= n; i++) {
            scanf("%d%s%d", &opt, s, &v);
            int tmp = ma;
            if(opt == ) {
                pii p = findx(s);
                if(p.first == - || vis[p.first] == ) {
                    cnt++;
                    vis[cnt] = ;
                    le[cnt] = re[cnt] = -;
                    ans[cnt] = v;
                    add(s, cnt, v);
                    if(tmp != -) re[tmp] = cnt;
                    le[cnt] = tmp;
                    ma = cnt;
                    num++;
                    printf("%d\n", v);
                }
                else {
　　　　　　　　　　　//更改的是第一个
                    if(p.first == mi) mi = re[mi];
                    if(num == ) mi = p.first;
                    printf("%d\n", p.second);
　　　　　　　　　　//注意这个！！！！更之前tmp有关，可能更改的是最后一个
                    if(p.first == ma) continue;
　　　　　　　　　　//链表操作。
                    if(le[p.first] != -) re[le[p.first] ] = re[p.first];
                    if(re[p.first] != -) le[re[p.first] ] = le[p.first];
　　　　　　　　　　//!!!!因为ma可能更改了，所以之前要特判一下p.first == ma.
                    if(tmp != -) re[tmp] = p.first;
                    le[p.first] = tmp;
                    re[p.first] = -;
                    ma = p.first;
                }
            }
            else {
                pii p = findx(s);
                if(p.first != - && vis[p.first] == ) {
                    if(v == ) printf("%d\n", p.second);
                    else if(v == -) {
                        if(le[p.first] != - && vis[p.first ] == ) printf("%d\n", ans[le[p.first] ]);
                        else printf("Invalid\n");
                    }
                    else {
                        if(re[p.first] != - && vis[p.first ] == ) printf("%d\n", ans[re[p.first] ]);
                        else printf("Invalid\n");
                    }
                }
                else printf("Invalid\n");
            }
            if(num > m) {
                num--;
　　　　　　　　//表示出容器。
                vis[mi] = ;
                int tm = re[mi];
                if(tm != -) le[tm] = -;
                re[mi] = -;
                mi = tm;
            }
        }
    }
    return ;
}
/*
1
8 3
0 0101010 1
0 0101011 2
1 0101010 1
0 1100000 3
0 0101011 -1
0 1111111 4
1 0101011 -1
1 0101010 0
*/

并查集加速链表操作

#include <bits/stdc++.h>
 
using namespace std;
const int maxn = 1e5 + ;
unordered_map<int, int> fa;
 
int findfa(int x) {
    if (!fa.count(x)) return x;
    return fa[x] = findfa(fa[x]);
}
 
int main() {
//    freopen("in.txt", "r", stdin);
//    freopen("out.txt", "w", stdout);
    int n, q;
    scanf("%d %d", &n, &q);
    int op, x;
    while (q--) {
        scanf("%d %d", &op, &x);
        if (op == ) {
            fa[x] = findfa(x + );
        } else {
            int ans = findfa(x);
            if (ans > n) ans = -;
            printf("%d\n", ans);
        }
    }
    return ;
}