本周的学习内容:
树状数组
用于对区间操作非常方便。
1001
求一个整数,二进制表示形式中最低位的1代表的整数。
就是树状数组中,求Lowbit的操作。
题解:
#include <iostream>
using namespace std;
int main()
{
int num;
cin>>num;
while(num!=0)
{
cout<<int(num&(-num))<<endl;
cin>>num;
}
return 0;
}
1002
树状数组的应用。
点操作,区间查询。
刚开始树状数组的建立过程,可以直接以树状数组的更新来建立。
#include <iostream>
#include <string>
#include <string.h>
using namespace std;
int last(int);
void Change(int,int);
int Sum(int);
int data[50001]= {0};
int N;
int main()
{
int T;
cin>>T;
for(int i=1; i<=T; i++)
{
cout<<"Case "<<i<<":"<<endl;
memset(data,0,sizeof(data));
cin>>N;
int num;
string order;
int First,Second;
for(int j=1; j<=N; j++)
{
cin>>num;
Change(j,num);
}
while(cin>>order)
{
if(order=="End")
break;
else
{
cin>>First>>Second;
if(order=="Add")
{
Change(First,Second);
}
if(order=="Sub")
{
Change(First,-Second);
}
if(order=="Query")
{
cout<<Sum(Second)-Sum(First-1)<<endl;
}
}
}
}
return 0;
}
int last(int x)
{
return x&-x;
}
void Change(int loc,int num)
{
while(loc<=N)
{
data[loc]+=num;
loc+=last(loc);
}
}
int Sum(int x)
{
int amount=0;
while(x!=0)
{
amount+=data[x];
x-=last(x);
}
return amount;
}
1003
树状数组的应用。
对时间复杂度的优化。
题解:
#include <iostream>
#include <string.h>
using namespace std;
int TreeData[100001]={0};
int N;
int Lowbit(int x)
{
return x&-x;
}
int Add(int x)
{
long long Sum=0;
while(x!=0)
{
Sum+=TreeData[x];
x-=Lowbit(x);
}
return Sum;
}
void Update(int x)
{
while(x<=N)
{
TreeData[x]++;
x+=Lowbit(x);
}
}
int main()
{
long long number;
long long int Sum=0;
while(cin>>N)
{
Sum=0;
memset(TreeData,0,sizeof(TreeData));
for(int i=0;i<N;i++)
{
cin>>number;
Update(number);
Sum+=(Add(N)-Add(number));
}
cout<<Sum<<endl;
}
return 0;
}
1004
气球涂色。
区间操作,区间内的单点查询。
差分树状数组。(用差分+前缀和也可以过,用树状数组只是对时间复杂度进行了优化)
题解:
#include <iostream>
using namespace std;
int main()
{
int N;
int Start,End;
while(cin>>N)
{
if(N==0)
break;
int data[N+1]={0};
for(int i=0;i<N;i++)
{
cin>>Start>>End;
data[Start]++;
data[End+1]--;
}
for(int i=1;i<=N;i++)
{
data[i]+=data[i-1];
}
for(int i=1;i<N;i++)
{
cout<<data[i]<<" ";
}
cout<<data[N]<<endl;
}
return 0;
}
1005
求逆序对(2)
离散化+树状数组。
离散化:
当所处理的数据较大时,无法开一个特别大的数组,因此,在不改变数据相对大小的条件下可以把数据进行相应的缩小。
例:
原数据:
{1234567,123456789,12345678,123456}
离散化后,可以把这四个数表示成:
{2,4,3,1}
因此,
题解:
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
const int Max=1e5+1;
int Data[Max];
int Sign[Max];
int N;
int Lowbit(int x)
{
return x&-x;
}
void Update(int x)
{
while(x<=N)
{
Sign[x]++;
x+=Lowbit(x);
}
}
long long Add(int x)
{
long long Sum=0;
while(x!=0)
{
Sum+=Sign[x];
x-=Lowbit(x);
}
return Sum;
}
int main()
{
while(cin>>N)
{
long long sum=0;
memset(Data,0,sizeof(Data));
for(int i=1; i<=N; i++)
{
cin>>Data[i];
Sign[i]=Data[i];
}
sort(Sign+1,Sign+1+N);
int m=unique(Sign+1,Sign+1+N)-Sign-1;
for(int i=1; i<=N; i++)
{
Data[i]=lower_bound(Sign+1,Sign+m+1,Data[i])-Sign;
}
memset(Sign,0,sizeof(Sign));
for(int i=1; i<=N; i++)
{
Update(Data[i]);
sum+=(Add(N)-Add(Data[i]));
}
cout<<sum<<endl;
}
return 0;
}