题目:
https://vjudge.net/problem/UVA-12345
题意:
给出一个有n个元素的数组,有以下两种操作:Q x y,求出区间[x,y)内不同元素的个数,M x y,把第x个元素的值修改为y。注意题目中的下标是从0开始的
思路:
我写了分块和带修改莫队两种做法。首先讲分块法,对于每个位置i,我们记录前一个相同元素的坐标,记为pre[i],然后分块,块内按pre数组排序,等到查询的时候区间[l,r)时,如果pre[i] < l,说明位置i的元素在查询区间内第一次出现,计数加1,如果pre[i] >= l,说明位置i的元素不是第一次出现,前面已经统计过了,故计数不变。按照这个思路,块内的pre数组是有序的,故可以二分快速确定第一次出现的元素的个数,不在完整块内的元素直接暴力统计即可。对于修改位置i的元素为v,我们可以发现,最多只需要找到四个值,皆可以完成修改,即找到位置i的前一个相同元素j和后一个相同元素的位置k,把k的前驱更新为j;找到位置i之前的第一个等于v的元素的位置x和之后第一个等于v的元素的位置y,把y的前驱更新为i,i的前驱更新为x,注意前驱不存在就更新为0,这样就完成了修改
#include <bits/stdc++.h>
using namespace std;
const int N = 50000 + 10, M = 1000000 + 10, INF = 0x3f3f3f3f;
int n, m;
int block, sz;
int pos[N], L[N], R[N], pre[N];
int last[M];
int a[N], b[N];
void reset(int x)
{
for(int i = L[x]; i <= R[x]; i++) pre[i] = b[i];
sort(pre + L[x], pre + R[x] + 1);
}
void init()
{
block = sqrt(n);
sz = n / block;
if(n % block) sz++;
for(int i = 1; i <= n; i++) pos[i] = (i-1) / block + 1;
for(int i = 1; i <= sz; i++)
{
L[i] = (i-1) * block + 1;
R[i] = i * block;
}
R[sz] = n;
memset(last, 0, sizeof last);
for(int i = 1; i <= n; i++)
{
b[i] = last[a[i]];
last[a[i]] = i;
}
for(int i = 1; i <= sz; i++) reset(i);
}
void update(int x, int v)
{
if(a[x] == v) return;
int idx = -1, idv = -1;
for(int i = x+1; i <= n; i++)
{
if(a[i] == a[x] && idx == -1) idx = i;
else if(a[i] == v && idv == -1) idv = i;
if(idx != -1 && idv != -1) break;
}
if(idv != -1) b[idv] = x, reset(pos[idv]);
idv = x;
for(int i = x-1; i >= 1; i--)
{
if(a[i] == a[x] && idx != -1) b[idx] = i, reset(pos[idx]), idx = -1;
if(a[i] == v && idv != -1) b[idv] = i, reset(pos[idv]), idv = -1;
if(idx == -1 && idv == -1) break;
}
if(idx != -1) b[idx] = 0, reset(pos[idx]);
if(idv != -1) b[idv] = 0, reset(pos[idv]);
a[x] = v;
}
int query(int l, int r)
{
int lb = pos[l], rb = pos[r];
int ans = 0;
if(lb == rb)
{
for(int i = l; i <= r; i++)
if(b[i] < l) ans++;
}
else
{
for(int i = l; i <= R[lb]; i++)
if(b[i] < l) ans++;
for(int i = L[rb]; i <= r; i++)
if(b[i] < l) ans++;
for(int i = lb + 1; i < rb; i++)
ans += lower_bound(pre + L[i], pre + R[i] + 1, l) - (pre + L[i]);
}
return ans;
}
int main()
{
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
init();
char ch;
int x, y;
for(int i = 1; i <= m; i++)
{
scanf(" %c%d%d", &ch, &x, &y);
if(ch == 'Q') printf("%d\n", query(x+1, y));
else update(x+1, y);
}
return 0;
}
带修改莫队和普通莫队其实差不多,普通莫队按两个关键字对询问排序,首先按左端点所在的块从小到大排序,否则按右端点排序。带修改莫队按三个关键字排序,第三个关键字是本次询问时前一个修改的位置,首先按左端点所在的块从小到大排序,否则按右端点所在的块从小到大排序,否则按第三个关键字直接排序。然后查询时,暴力还原修改到本次查询时的状态,其他的就都一样了。代码中的vis[i]为1时代表这个值被统计在答案中,否则不是
#include <bits/stdc++.h>
using namespace std;
const int N = 50000 + 10, M = 1000000 + 10;
struct node
{
int l, r, id, pre;
}g[N];
struct
{
int x, v, o;
}q[N];
int n, m;
int block, tot, top, val;
int a[N], last[N], pos[N], ans[N], num[M];
bool vis[N];
bool cmp(node a, node b)
{
// return (pos[a.l] < pos[b.l]) || (pos[a.l]==pos[b.l] && pos[a.r] < pos[b.r])
// || (pos[a.l] == pos[b.l] && pos[a.r] == pos[b.r] && a.pre < b.pre);
if(pos[a.l] == pos[b.l] && pos[a.r] == pos[b.r]) return a.pre < b.pre;
else if(pos[a.l] == pos[b.l]) return pos[a.r] < pos[b.r];
else return pos[a.l] < pos[b.l];
}
void init()
{
//block = sqrt(n);
block = 1300;//设为1300比sqrt(n)块,蜜汁。。。
for(int i = 1; i <= n; i++) pos[i] = (i-1) / block + 1;
tot = top = 0;
}
void update(int x)
{
if(vis[x])
{
if(--num[a[x]] == 0) val--;
}
else
{
if(++num[a[x]] == 1) val++;
}
vis[x] ^= 1;
}
void restore(int x, int v)
{
if(vis[x])
{
update(x); a[x] = v; update(x);
}
else a[x] = v;
}
void work()
{
sort(g + 1, g + 1 + tot, cmp);
val = 0;
int l = 1, r = 0, now = 0;
for(int i = 1; i <= tot; i++)
{
while(now < g[i].pre) ++now, restore(q[now].x, q[now].v);
while(now > g[i].pre) restore(q[now].x, q[now].o), --now;
while(r < g[i].r) update(++r);
while(r > g[i].r) update(r--);
while(l < g[i].l) update(l++);
while(l > g[i].l) update(--l);
ans[g[i].id] = val;
}
for(int i = 1; i <= tot; i++) printf("%d\n", ans[i]);
}
int main()
{
scanf("%d%d", &n, &m);
init();
for(int i = 1; i <= n; i++) scanf("%d", &a[i]), last[i] = a[i];
char ch;
int x, y;
for(int i = 1; i <= m; i++)
{
scanf(" %c%d%d", &ch, &x, &y);
x++;
if(ch == 'Q') g[++tot].l = x, g[tot].r = y, g[tot].id = tot, g[tot].pre = top;
else q[++top].x = x, q[top].v = y, q[top].o = last[x], last[x] = y;
}
work();
return 0;
}