[HEOI2012]采花(思维 + 离线 + 树状数组)

题目链接

分析

我只能说太妙了…

离线 + 树状数组

参考题解:

这个题要和《HH的项链》做对比;
关键就是为什么要离线处理,以及 怎么对区间进行排序和维护;

[SDOI2009]HH的项链:
这个题求的是区间中不同数的个数,假设我们从前往后遍历;
遍历到a[i]时,对于前面的x(x=a[i]),我们要选一个做代表;
不妨选a[i]作为代表,则我们应该消除前面所有x的影响;
我们只用记录每个数出现的上个位置,然后每次更新就行;
然后我们发现这个过程是不可逆的,如果先处理r大的再处理r小的那么就会有问题;
因此我们需要对r从小到大排序;

[HEOI2012]采花:
这个题不同,这个题只有区间中的数出现>=2时才会算一次贡献;
这个问题就很奇怪…我们可以维护一下相邻相同数的关系;
即记录一下a[i]这个数下一次出现的下标ne[a[i]];
从前往后考虑:(也是用一个指针now迭代)
对于一个数a[now],如果当前询问的区间是[l,r],now<l;
即a[now]在询问区间外,那么a[now]的下一个数ne[a[now]]在这个区间不产生贡献;
我们可以update(ne[a[now]],-1);
但是对于ne[ne[a[now]]],是有可能产生贡献的,因此我们update(ne[ne[a[now]]],1);
由此可以知道我们这个now指针和区间l是有关的,now从前往后遍历l,我们可以按l从小到大排区间;
此时我们query®,这样区间内出现过一次的数都没有贡献,出现过多次的数我们也只算了一次(在第二次出现的时候打了一次标记)

这里为啥不用query(l)?
其实我们在遍历now的时候,a[now]第一次出现是没有算贡献的,当l变大;
now在遍历时,同时会更新,消除区间外的数的影响,所以query(l)是0;

代码

#include<iostream>
#include<queue>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<map>
#include<algorithm>
#include<deque>
#include<stack>
#include<set>
#include <unordered_map>
#include<math.h>
#include<string.h>
#define IOS ios::sync_with_stdio(false),cin.tie(0);
using namespace std;
 
#define pb push_back
#define coutl cout<<"------------"<<endl;
#define fi first
#define se second
  
#define ire(x) scanf("%d",&x)
#define iire(a,b) scanf("%d %d",&a,&b)
#define lre(x) scanf("%lld",&x)
#define llre(a,b) scanf("%lld %lld",&a,&b)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define endl "\n"
#define PI acos(-1.0)

typedef long long ll;
typedef unsigned long long ull;
      
typedef pair<int, int> PII;
typedef pair<double, int> PDI;
typedef pair<ll, ll> PLL;
typedef pair<double, double> PDD;
typedef pair<double, pair<int, double>> PDID;
typedef pair<char, char> PCC;
typedef pair<char, pair<int, int> > PCII;
typedef pair<int, pair<int, int> > PIII;
typedef pair<int, pair<int, pair<int, int> > > PIIII;
typedef pair<ll, pair<int, int> > PLII;
 
const int maxn = 3e6 + 7;
const int N = 2010 + 7;
const int M = 1e6 + 7;
const int mod = 998244353;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const double pi = acos(-1);
const double eps = 1e-8;
  
ll gcd(ll a,ll b) {return b==0 ? a : gcd(b,a%b);}
ll lcm(ll a,ll b) {return a*b / gcd(a,b);}
ll qmi(ll a,ll b,ll p) {ll ans = 1; while(b) { if(b & 1) ans = ans * a % p; a = a * a % p; b >>= 1; } return ans;}
int lowbit(int x) {return x & (-x);}

int n,c,m;
int a[maxn];
int ne[maxn];
int tr[maxn];
int vis[maxn];
int ans[maxn];
int pos[maxn];

struct node
{
	int l,r;
	int id;
}no[maxn];

bool cmp(node no1,node no2)
{
	return no1.l < no2.l;
}

void update(int x,int v)
{
	while(x <= n)
	{
		tr[x] += v;
		x += lowbit(x);
	}
}

int query(int x)
{
	int ans = 0;
	while(x)
	{
		ans += tr[x];
		x -= lowbit(x);
	}
	return ans;
}

int main()
{
	scanf("%d %d %d",&n,&c,&m);
	for(int i=1;i<=n;i++) ire(a[i]);
	
	for(int i=1;i<=m;i++)
	{
		int l,r;
		iire(l,r);
		no[i] = {l,r,i};
	}
	sort(no+1,no+1+m,cmp);
	
	//从后往前处理每个数的下个数的位置
	for(int i=n;i>=1;i--)
	{
		ne[i] = pos[a[i]];
		pos[a[i]] = i;
	}
	
	//初始化,每个数第二次出现的位置打标记
	for(int i=1;i<=n;i++)
		if(!vis[a[i]] && ne[i]) update(ne[i],1), vis[a[i]] = 1;
	
	int now = 0;
	for(int i=1;i<=m;i++)
	{
		while(now < no[i].l)	//在区间外
		{
			if(ne[now])
			{
				update(ne[now],-1);
				if(ne[ne[now]]) update(ne[ne[now]],1);
			}
			now++;
		}
		
		ans[no[i].id] = query(no[i].r);
	}
	
	for(int i=1;i<=m;i++) cout<<ans[i]<<'\n';
	
	return 0;
}



  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值