树状数组

Stars POJ2352

 

题意：求每个等级的数量，等级0-N-1;

划分等级：有一个星星Ａ，横纵坐标均小于等于星星Ａ的横纵坐标的星星有多少个，则星星Ａ的等级为几．

分析：该题给出的样例的纵坐标均是从小到大排列，所以我们只需要查找横坐标就行了，统计出现的小于等于该横坐标的横坐标有多少个，则该星星为等级几．用vis[getsun(x+1)]来统计该等级的数量．注意等级是0-n-1范围，而c[]数组下标是从1开始的，如果用getsum(x)的话为时间超限．更新也是，update(x+1).

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN = 32005;
int c[MAXN],vis[MAXN];
int n;
int lowb(int i)
{
return i&(-i);
}
void update(int i,int v)
{
    while(i<=32005)
    {
        c[i]+=v;
        i+=lowb(i);
    }
}
int getsum(int i)
{
    int ans=0;
    while(i>0)
    {
        ans+=c[i];
        i-=lowb(i);
    }
    return ans;
}
int main()
{

    int i,a,b;
    scanf("%d",&n);
    memset(c,0,sizeof(c));
    memset(vis,0,sizeof(vis));
    for(i=1;i<=n;i++)
    {
        scanf("%d %d",&a,&b);
        a++;
        vis[getsum(a)]++;
        update(a,1);
    }
    for(i=0;i<n;i++)
    {
        printf("%d\n",vis[i]);
    }
    return 0;
}

 Ｃows  POJ2481

这道题求的是比该牛强壮的有几只牛嘻嘻嘻

分析：给了该牛吃草的范围（x,y）,其中牛A的吃草范围(x0,y0)，如果x0<=x并且y0>=y并且y0-x0>y-x，则牛Ａ比该牛强壮；

我们只需要用个结构体把该牛的范围和下标存下来，然后按照y的范围排个序即可．遍历的时候，如果前面出现了小于x的，那么就比它强壮，统计一下它前面的并且x小于它的，不存在y的烦恼啦，y是按照从大到小排序，所以一定是包含在里面的啦～

注意x的范围是从0开始，所以getsum()和update()的时候要x+1，否则时间超限，和上面那道题是一样的

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN = 100005;
int c[MAXN],ans[MAXN];
int n;
struct A
{
    int a;
    int b;
    int vis;
}t[100005];
bool cmp(A p,A q)
{

     if(p.b==q.b)
    return p.a<q.a;
    return p.b>q.b;

}
int lowb(int i)
{
return i&(-i);
}
void update(int i,int v)
{
    while(i<=n)
    {
        c[i]+=v;
        i+=lowb(i);
    }
}
int getsum(int i)
{
    int ans=0;
    while(i>0)
    {
        ans+=c[i];
        i-=lowb(i);
    }
    return ans;
}
int main()
{

    int i;
    while(~scanf("%d",&n))
    {
    if(n==0)
        break;
    memset(c,0,sizeof(c));
    memset(ans,0,sizeof(ans));
    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&t[i].a,&t[i].b);
        t[i].vis=i;
    }
    sort(t+1,t+1+n,cmp);
    for(i=1;i<=n;i++)
    {
      if(t[i].a==t[i-1].a&&t[i].b==t[i-1].b)
      {
          ans[t[i].vis]=ans[t[i-1].vis];
      }
      else
      ans[t[i].vis]=getsum(t[i].a+1);
      update(t[i].a+1,1);
    }
    for(i=1;i<=n;i++)
    {
        if(i!=n)
        printf("%d ",ans[i]);
        else
        printf("%d\n",ans[i]);
    }
    }
    return 0;
}

 Japan POJ3067

题意：求交点数

分析：可以在本子上画一下．用结构体存每一组的数据，x按从小到大排序,如果x相等则按y从小到大排序，然后求它前面的但y比它大的个数，能联想到其实就是求逆序对的个数．用ans记录交点数．getsum()本来求的是小于等于它的个数，但是这里可以用ans+=getsum(m)-getsum(y)来表示

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long ll;
const int MAXN = 2005;
struct A{
int a,b;
}t[MAXN*MAXN];
ll n,m,k,c[MAXN];

bool cmp(A p,A q)
{
    if(p.a==q.a)
        return p.b<=q.b;
    else
        return p.a<q.a;
}
int lowbit(int i)
{
    return i&(-i);
}
void update(int i,int v)
{
    while(i<=m)
    {
        c[i]+=v;
        i+=lowbit(i);
    }
}
ll getsum(int i)
{
    ll ans=0;
    while(i>0)
    {
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
int main()
{
    int T,i,j;
    scanf("%d",&T);
    for(i=1;i<=T;i++)
    {
        ll an=0;
        scanf("%lld %lld %lld",&n,&m,&k);
        memset(c,0,sizeof(c));
        for(j=1;j<=k;j++)
        {
            scanf("%d %d",&t[j].a,&t[j].b);
        }
        sort(t+1,t+1+k,cmp);
        for(j=1;j<=k;j++)
        {
            update(t[j].b,1);
            an+=(getsum(m)-getsum(t[j].b));

        }
        printf("Test case %d: %lld\n",i,an);
    }
    return 0;
}

Ｃolor the ball HDU1556

题意就不说了，线段树里面有

附上用树状数组做的代码

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
const int MAXN = 100005;
int c[MAXN],n;
int lowbit(int i)
{
    return i&(-i);
}
void update(int i,int v)
{
    while(i<=n)
    {
        c[i]+=v;
        i+=lowbit(i);
    }
}
int getsum(int i)
{
    int ans=0;
    while(i>0)
    {
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
int main()
{
    int a,b,i;
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)
            break;
        memset(c,0,sizeof(c));
        for(i=1;i<=n;i++)
        {
            scanf("%d %d",&a,&b);
            update(a,1);
            update(b+1,-1);
        }
        for(i=1;i<=n;i++)
        {
            if(i!=n)
                printf("%d ",getsum(i));
            else
                printf("%d\n",getsum(i));
        }
    }
    return 0;
}