前言
一天一个专题,还行。
一直都有keep住树状数组的正确姿势,这次也主要是练习了一些变式题目,一些处理比记住板子更加重要,只要知道这东西的问法和用途往上套才行。
之前在ybtoj上面有写过六道模板题,比今天的题目更难,可以回去看看,特别是带有区间修改的。
对于树状数组在区间中的用途,最经典的就是区间加减和差和逆序对,有时候需要将问题转化,得到模型,后面的题目中会具体展现。最基础直白的模板就不放了。
例题
1. poj2481 Cows
分析
题意就是求一个区间被几个区间“覆盖”(完全重叠不算)。
一开始想开两个树状数组的,但是没办法实现题目的操作。所以就对区间头升序排列,区间头相等的区间尾降序排列。然后实时加入树状数组,因为这样的排序,所以能覆盖他的都在他前面,统计前面有多少个区间尾比他大的,就是覆盖的。注意相等的情况,个数跟前面那个一样就行。
上代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
struct node
{
int x,y,p;
}a[100010];
int n,c[100010],ans[100010],mx;
int cmp(node l,node r)
{
if(l.x==r.x) return l.y>r.y;
else return l.x<r.x;
}
int cmp2(node l,node r)
{return l.p<r.p;}
int lowbit(int x){return x&(-x);}
int change(int x,int v)
{for(int i=x;i<=mx;i+=lowbit(i)) c[i]+=v;}
int getsum(int x)
{
int s=0;
for(int i=x;i;i-=lowbit(i)) s+=c[i];
return s;
}
int main()
{
scanf("%d",&n);
while(n!=0)
{
for(int i=1;i<=n;i++)
{
scanf("%d%d",&a[i].x,&a[i].y);
a[i].x++;a[i].y++;
a[i].p=i;
mx=max(mx,a[i].y);
}
sort(a+1,a+n+1,cmp);
for(int i=1;i<=n;i++)
{
if(i!=1&&a[i].x==a[i-1].x&&a[i].y==a[i-1].y)
{
ans[a[i].p]=ans[a[i-1].p];
}
else ans[a[i].p]=getsum(mx)-getsum(a[i].y-1);
change(a[i].y,1);
}
sort(a+1,a+n+1,cmp2);
for(int i=1;i<=n-1;i++) printf("%d ",ans[i]);
cout<<ans[n]<<endl;
scanf("%d",&n);
memset(c,0,sizeof(c));
memset(ans,0,sizeof(ans));
mx=0;
}
return 0;
}
2. poj3067 Japan
分析
手动模拟一下样例,很容易发现这是一道跟逆序对有关的题目。
我们将西海岸的城市作为数组的下标,东海岸的城市作为数组值,统计每个数加进来对逆序对的贡献。
具体实现就需要对西海岸城市升序排列,确保加进来的数数组下标不减。然后判断有多少个数 “比他先进来还比他大”。
最后吐槽一下poj,居然不给用“%lld”,人家CSP可是只能用这个的啊。。
上代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
struct node
{
int x,y;
}a[1000100];
int t,n,m,k;
int c[1000100];
ll ans;
int cmp(node l,node r)
{
if(l.y==r.y) return l.x<r.x;
else return l.y<r.y;
}
int lowbit(int x){return x&(-x);}
void change(int x,int v)
{for(int i=x;i<=n;i+=lowbit(i)) c[i]+=v;}
ll getsum(int x)
{
ll s=0;
for(int i=x;i;i-=lowbit(i)) s+=c[i];
return s;
}
int main()
{
scanf("%d",&t);
for(int i=1;i<=t;i++)
{
memset(c,0,sizeof(c));
ans=0;
scanf("%d%d%d",&n,&m,&k);
/*西海岸作为下标,东海岸作为值*/
for(int j=1;j<=k;j++)
{
scanf("%d%d",&a[j].x,&a[j].y);
}
sort(a+1,a+k+1,cmp);
for(int j=1;j<=k;j++)
{
change(a[j].x,1);
ans+=j-getsum(a[j].x);//逆序对
}
printf("Test case %d: %I64d\n",i,ans);//poj这个才能过
//printf("Test case %d: %lld\n",i,ans);
}
return 0;
}
3. poj2155 矩阵
分析
发现这道题是取反,而不是累加,所以简单的矩阵加矩阵求和套进去是不可行的。
但是为什么他是矩阵取反单点求和呢,感觉有点奇怪是不是。
考虑取反运算的性质,取反偶数次是0,奇数次是1。
首先考虑一维的情况,我要一段区间取反,假设是 [l, r]。那么我只需要c[l]+1,c[r+1]+1,假设查询 k 的时候,只需要查询前 k 的和 mod 2 的结果即可。
然后我们用容斥轻松拓展到二维。
这里红色的就是上面需要标记的点,标记后表示以这个点为右下角的子矩阵内的点全部取反一次。这里绿色代表取反一次,紫色是取反了两次,黄色是取反了四次。最后实际进行取反操作的就是要求取反的子矩阵内的点。(本段转载,侵删)
最后这个问题变成了单点修改,区间查询的二维树状数组。
上代码
再次吐槽poj奇奇怪怪的输入输出。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int t;
int n,m;
int c[1010][1010];
int lowbit(int x){return x&(-x);}
void update(int x,int y,int k)
{
for(int i=x;i<=n;i+=lowbit(i))
{
for(int j=y;j<=n;j+=lowbit(j))
{
c[i][j]+=k;
}
}
}
long long getsum(int x,int y)
{
long long s=0;
for(int i=x;i;i-=lowbit(i))
{
for(int j=y;j;j-=lowbit(j))
{
s+=c[i][j];
}
}
return s;
}
int main()
{
cin>>t;
while(t--)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
char op;
cin>>op;
if(op=='Q')
{
int x,y;
scanf("%d%d",&x,&y);
cout<<getsum(x,y)<<endl;
printf("%d\n",getsum(x,y)%2);
}
else
{
int x1,y1,x2,y2;
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
update(x1,y1,1);
update(x1,y2+1,1);
update(x2+1,y1,1);
update(x2+1,y2+1,1);
}
}
printf("\n");
memset(c,0,sizeof(c));
}
return 0;
}
写在后面
希望CSP的时候我能用上这个好东西。