NC19427 换个角度思考——思维+树状数组/线段树

题目链接

思路

这道题思路挺新奇的，树状数组维护前缀和。先存下输入的每个询问，再按照k的值从小到大排序。对于a数组开一个结构体数组，存下下标和值，按照值从小到大排序。对于每一个$k_i$，将小于等于$k_i$的值的位置add 1。然后这个询问的答案就是ask® -ask(l-1)。因为已经按照k从小到大排序了，所以之前插入的k对之后没有影响

AC代码

#include<bits/stdc++.h>

#define lp p<<1
#define rp p<<1|1
using namespace std;
const int N = 1e5+9;
int tr[N],ans[N];
struct node
{
	int id,num;
	bool operator<(const node &t) const
	{
		return num<t.num;
	}
}a[N];
struct Query
{
	int l,r,k,id;
	bool operator<(const Query &t) const
	{
		return k<t.k;
	}
}q[N];
int lowbit(int x) { return x&(-x);}
void add(int pos,int c)
{
	for(int i=pos; i<N; i+=lowbit(i)) tr[i]+=c;
}
int ask(int p)
{
	int res=0;
	for(int i=p; i; i-=lowbit(i)) res+=tr[i];
	return res;
}
int main()
{
	int n,m;
	scanf("%d %d",&n,&m);
	for(int i=1; i<=n; i++) scanf("%d",&a[i].num),a[i].id=i;
	sort(a+1,a+n+1);
	for(int i=1; i<=m; i++) scanf("%d %d %d",&q[i].l,&q[i].r,&q[i].k),q[i].id=i;
	sort(q+1,q+m+1);
    for(int i=1,j=1; i<=m; i++)
    {
    	int num=q[i].k;
    	//puts("debug");
    	while(j<=n && a[j].num<=num) add(a[j++].id,1);
    	ans[q[i].id]=ask(q[i].r)-ask(q[i].l-1);
	}
	for(int i=1; i<=m; i++) printf("%d\n",ans[i]);
	return 0;
 }