CF1000F One Occurrence(莫隊)

給定一個長度爲n序列,m個詢問,每次詢問給定一個區間[l,r],如果這個區間裏存在只出現一次的數,輸出這個數(如果有多個就輸出任意一個),沒有就輸出0,n,m<=5*10^5

輸入格式:

第一行一個整數n
接下來1行n個小於5*10^5
的正整數,即序列
下面一行一個整數m
在下面m行,每行兩個整數l,r
6
1 1 2 3 2 4
2
2 6
1 2

莫隊板子題,分塊優化查詢(記錄只出現一次的數在哪個塊中出現)暴力分塊查找
時間複雜度:msqrt(n) + nsqrt(n)

#include <iostream>
#include <math.h>
#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e5 + 5;
struct node {
    int l, r, id;
}q[maxn];
int col[maxn], cnt[maxn], pos[maxn + 5];
int L[maxn + 5], R[maxn + 5];
int res[maxn];
inline bool cmp(node a, node b) {
    if(pos[a.l] != pos[b.l]) {
        return pos[a.l] < pos[b.l];
    }
    if(pos[a.l] & 1) {
    	return a.r < b.r;
    }
    return a.r > b.r;
    
}
void add(int x) {
    cnt[col[x]]++;
    if (cnt[col[x]] == 1) { //如果只出現一次
        res[pos[col[x]]]++; //這個數所在的塊+1
    } else if (cnt[col[x]] == 2){ //如果出現不止一次
        res[pos[col[x]]]--; //這個數所在的塊-1
    }
}
void del(int x) {
    cnt[col[x]]--;
    if (cnt[col[x]] == 1) {//如果只出現一次
        res[pos[col[x]]]++;//這個數所在的塊+1
    } else if (cnt[col[x]] == 0) { //如果不在這個範圍內
        res[pos[col[x]]]--; //這個數所在的塊-1
    }
}
int read() {
    int ans = 0;
    char ch = getchar();
    while (ch < '0' || ch > '9') ch = getchar();
    while (ch >= '0' && ch <= '9') {
        ans = ans * 10 + ch - '0';
        ch = getchar();
    }
    return ans;
}

int ans[maxn];
int main() {
    int n, m;
    int block = sqrt(maxn);
    int num = maxn / block + (maxn % block != 0);
    for (int i = 1; i <= maxn; i++) {
        pos[i] = (i - 1) / block + 1;
    }
    for (int i = 1; i <= num; i++) {
        L[i] = (i - 1) * block + 1;
        R[i] = i * block;
    }
    R[num] = maxn;
    n = read();
    for (int i = 1; i <= n; i++) {
        col[i] = read();
    }
    m = read();
    for (int i = 1; i <= m; i++) {
        q[i].l = read(); q[i].r = read();
        q[i].id = i;
    }
    sort (q + 1, q + m + 1, cmp);
    int l = 1, r = 0;
    for (int i = 1; i <= m; i++) { //莫隊板子
        while(l < q[i].l) {
            del(l++);
        }
        while(r < q[i].r) {
            add(++r);
        }
        while(l > q[i].l) {
            add(--l);
        }
        while(r > q[i].r) {
            del(r--);
        }
        for (int j = 1; j <= num; j++) { //暴力查詢只出現一次的數在哪個塊中出現
        	if (res[j]) {
        		for (int k = L[j]; k <= R[j]; k++) { //查詢出現的數字
        			if (cnt[k] == 1) {
        				ans[q[i].id] = k; break;
                    }
                }
        		break;
            }
        }
    }
    for (int i = 1; i <= m; i++) {
        printf("%d\n", ans[i]);
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章