训练日志9.19

P1972 [SDOI2009] HH的项链

这题主要是写的莫队的，但是可能数据加强了过不了（变形）

（还有树状数组和分块的写法应该差不多，留到下次写）

1.线段树

按照右端点进行排序，然后扫描要求区间的右端点，然后进行修改线段树，如果这个数之前出过，则减掉，并同时记录此时的区间的结果。

#include<bits/stdc++.h>
#define ls id<<1
#define rs id<<1|1
using namespace std;
const int N=5e6;
int a[N],p[N],ans[N];
int n,m;
struct qu{
    int l,r;
    int id;
    bool operator<(const qu &w){
        return r<w.r;
    }
}q[N];

struct tree{
    int l,r;
    int sum;
}tr[N];
void pushup(int id){
    tr[id].sum=tr[ls].sum+tr[rs].sum;
}
void build(int id,int l,int r){
    tr[id]={l,r,1};
    if(l==r)return ;
    int mid=(l+r)>>1;
    build(ls,l,mid);
    build(rs,mid+1,r);
    pushup(id);
}
void mod(int id,int x){
    if(tr[id].l==x&&tr[id].r==x){
        tr[id].sum=0;
        return ;
    }
    int mid=(tr[id].l+tr[id].r)>>1;
    if(x<=mid)mod(ls,x);
    if(x>mid)mod(rs,x);
    pushup(id);
}
int query(int id,int l,int r){
    if(l>tr[id].r||r<tr[id].l)return 0;
    if(l<=tr[id].l&&tr[id].r<=r){
        return tr[id].sum;
    }
    
    return query(ls,l,r)+query(rs,l,r);
}
vector<int>v[N];
signed main(){
    
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d %d",&q[i].l,&q[i].r);
        q[i].id=i;
    }
    sort(q+1,q+m+1);
    for(int i=1;i<=m;i++){
        v[q[i].r].push_back(i);
    }
    build(1,1,n);
    for(int i=1;i<=n;i++){
        if(p[a[i]])mod(1,p[a[i]]);
        p[a[i]]=i;
        for(auto x:v[i]){
            ans[q[x].id]=query(1,q[x].l,q[x].r);
        }
    }
    for(int i=1;i<=m;i++){
        printf("%d\n",ans[i]);
    }
    return 0;
}

2.莫队

按照书上的写法写的，主要是为了搞下莫队

#include<bits/stdc++.h>
using namespace std;
const int N=6e6;
inline int read(){
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
	return s*w;
}
inline void write(int x){
	if(x==0){putchar('0');return;}
	int len=0,k1=x,c[10005];
	if(k1<0)k1=-k1,putchar('-');
	while(k1)c[len++]=k1%10+'0',k1/=10;
	while(len--)putchar(c[len]);
}
int a[N];
int pos[N],res[N],cnt[N];
int block;
int ans=0;
struct qu{
	int l,r;
	int id;
	bool operator<(const qu &w)const{
		if(pos[l]!=pos[w.l])return pos[l]<pos[w.l];
		if(pos[l]&1)return r>w.r;
		return r<w.r;
	}
}q[N];
//bool cmp(qu a,qu 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[a[x]]++;
	if(cnt[a[x]]==1)ans++;
}
void del(int x){
	cnt[a[x]]--;
	if(cnt[a[x]]==0)ans--;
}
signed main(){

	int n,m;
	n=read();
	block=sqrt(n);
	for(int i=1;i<=n;i++){
		a[i]=read();
		pos[i]=(i-1)/block+1;
	}
	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);
	int L=1,R=0;
	for(int i=1;i<=m;i++){
		while(L<q[i].l)del(L++);
		while(R>q[i].r)del(R--);
		while(L>q[i].l)add(--L);
		while(R<q[i].r)add(++R);
		res[q[i].id]=ans;
	}
	for(int i=1;i<=m;i++){
		write(res[i]);
		putchar('\n');
	}
	return 0;
}