给出q个询问,询问一段区间里面的不同元素的个数有多少个。
离线做,用树状数组。
设树状数组的意义是:1--pos这个段区间的不用元素的种类数。怎么做?就是add(pos,1);在这个位置中+1,就是说这个位置上元素种类+1。
然后先把询问按R递增的顺序排序。因为这里是最优的,我每次尽量往R靠,使得查询不重不漏。
什么意思呢?
就是假如有:2、1、3、5、1、7的话。
一开始的[1,4]这段数字全部压进树状数组,用个数组book[val],表示val这个元素出现的最右的位置,因为我们需要删除重复的,也是要尽量往右靠。到达pos=5这个位置的时候,注意了,因为1是出现过的book[1] = 2,所以我们要做的是把2这个位置出现元素的种类数-1,就是add(book[1],-1)。然后把第五个位置出现的元素种类数+1,就是add(5,1)。为什么呢?因为你尽量把种类往右靠,因为我们的R是递增的,这样,你使得查询[4,6]成为可能,因为我那个1加入来了,而不是一直用pos=2那个位置的1,再者,查询[4,7]的话,一样的意思,因为中间的1进来了。所以我们因为尽量往右靠,毕竟我们都把query按R排序了。
还有这个只能离线,一直预处理ans[i]表示第i个询问的ans。更新到[4,7]后,查询[1,2]已经不可能了,因为很明显,pos=2这个位置已经被删除了。
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;
#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <string>
const int maxn = 1e6+20;
int book[maxn];
int c[maxn];//树状数组,多case的记得要清空
int n; //我只需记录这个位置是不是新元素,不用离散
int lowbit (int x)//得到x二进制末尾0的个数的2次方 2^num
{
return x&(-x);
}
void add (int pos,int val)//在第pos位加上val这个值
{
while (pos<=n) //n是元素的个数
{
c[pos] += val; pos += lowbit(pos);
}
return ;
}
int get_sum (int pos) //求解:1--pos的总和
{
int ans = 0;
while (pos)
{
ans += c[pos]; pos -= lowbit(pos);
}
return ans;
}
int a[maxn];
struct node
{
int L,R,id;
bool operator < (const struct node &rhs) const
{
if (R != rhs.R) return R < rhs.R;
else return L < rhs.L;
}
}query[maxn];
int ans[maxn];
void work ()
{
scanf("%d",&n);
for (int i=1;i<=n;++i) scanf("%d",&a[i]);
//树状数组含义:1--pos间的不同元素种类数
int q; scanf("%d",&q);
for (int i=1;i<=q;++i)
{
scanf("%d%d",&query[i].L,&query[i].R);
query[i].id = i; //记录ans
}
sort(query+1,query+1+q);
int cur = 1;
//book[val]的含义,val这个元素出现的最右边的位置
for (int i=1;i<=q;++i)
{
for (int j=cur;j<=query[i].R;++j)
{
if (book[a[j]]) //这个元素在之前位置出现过,我们尽量往右,所以先删除那个位置的种类数,更新现在这个位置的种类数。因为这样的话,使得查询[4,5]是可能的
add(book[a[j]],-1); //del 这个位置
book[a[j]]=j; //更新这个位置的最右值
add(j,1); //这个位置出现了新元素
}
cur = query[i].R+1;
ans[query[i].id] = get_sum(query[i].R) - get_sum(query[i].L-1);
}
for (int i=1;i<=q;++i)
printf ("%d\n",ans[i]);
}
int main()
{
#ifdef local
freopen("data.txt","r",stdin);
#endif
work();
return 0;
}
View Code
HDU 3333 Turing Tree
一样的,只不过是求元素的和。。把add那里的val改成a[i]即可,幻想它出现了val次,一统计,就是了。
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;
#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <string>
const int maxn = 30000 + 20;
LL c[maxn];//树状数组,多case的记得要清空
int n;
LL a[maxn];
struct node
{
int L,R,id;
bool operator < (const struct node &rhs) const
{
return R < rhs.R;
}
}query[2*100000];
LL ans[2*100000];
map<LL,int>book;
int lowbit (int x)//得到x二进制末尾0的个数的2次方 2^num
{
return x&(-x);
}
void add (int pos,LL val)//在第pos位加上val这个值
{
while (pos<=n) //n是元素的个数
{
c[pos] += val; pos += lowbit(pos);
}
return ;
}
LL get_sum (int pos) //求解:1--pos的总和
{
LL ans = 0;
while (pos)
{
ans += c[pos]; pos -= lowbit(pos);
}
return ans;
}
void init ()
{
memset(c,0,sizeof c);
book.clear();
return ;
}
void work ()
{
init();
scanf("%d",&n);
for (int i=1;i<=n;++i) scanf("%I64d",&a[i]);
int q;
scanf ("%d",&q);
for (int i=1;i<=q;++i)
{
scanf("%d%d",&query[i].L,&query[i].R);
query[i].id = i;
}
sort(query+1,query+1+q);
int cur=1;
for (int i=1;i<=q;++i)
{
for (int j=cur;j<=query[i].R;++j)
{
int t = book[a[j]];
if (t)
{
add(t,-a[j]);
}
book[a[j]]=j;
add(j,a[j]);
}
cur = query[i].R+1;
ans[query[i].id] = get_sum(query[i].R) - get_sum(query[i].L-1);
}
for (int i=1;i<=q;++i)
{
printf ("%I64d\n",ans[i]);
}
}
int main()
{
#ifdef local
freopen("data.txt","r",stdin);
#endif
int t;
scanf ("%d",&t);
while (t--) work();
return 0;
}
View Code