题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1556
题意:有n个数,编号为1到n,每次一个操作 a b,表示区间[a,b]内的数都记一次,最后输出每个数记过几次。
思路:n的范围是10W,平方模拟的算法会超时。我们要优化每次对区间的操作,如果对区间[3,5]操作一次,x[3]++,x[6]--。
那么我们观察前六个数的区间和 0 0 1 1 1 0 ,前缀和的状态正好可以表示我们刚才的操作,而且去更新前缀和可以借助树状数组优化至logn。
树状数组专题链接:http://blog.csdn.net/chy20142109/article/details/50673749
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
using namespace std;
#define rep(i,j,k) for(int i = j; i <= k; i++ )
#define Rrep(i,j,k) for(int i = j; i >= k; i-- )
#define Clean(x,y) memset(x,y,sizeof(x))
const int maxn = 100008;
int n;
int sum[maxn];
int lowbit(int x)
{
return x&(-x);
}
int query(int x)
{
int ans = 0;
while( x )
{
ans += sum[x];
x-=lowbit(x);
}
return ans;
}
void add(int x,int k)
{
while( x<=n )
{
sum[x]+=k;
x+=lowbit(x);
}
}
int main()
{
while(cin>>n)
{
if ( !n ) break;
Clean(sum,0);
int a,b;
rep(i,1,n)
{
scanf("%d %d",&a,&b);
add(a,1);
add(b+1,-1);
}
rep(i,1,n) printf("%d%c",query(i),(i==n?'\n':' ' ));
}
return 0;
}