POJ1007(DNA Sorting)

题目链接

求逆序数的题。因为只有4种元素，所以可以直接统计排在某个元素前面且比它小的元素个数（用4个变量记录已出现的A,C,G,T的个数），复杂度为O(N)。若元素种类很多，可以使用树状数组进行类似基数排序的统计。

#include <stdio.h>
#include <stdlib.h>
#define N 55
#define M 105
struct node
{
    char s[N];
    int d;
}node[M];
int a,c,g,t,len;
int cal(char s[])
{
    int i,sum;
    a=c=g=t=0;
    sum=0;
    for(i=0;i<len;i++)
    {
        switch(s[i])
        {
            case 'A':   a++,sum+=c+g+t; break;
            case 'C':   c++,sum+=g+t;   break;
            case 'G':   g++,sum+=t; break;
            case 'T':   t++;    break;
        }
    }
    return sum;
}
int cmp(const void*a,const void*b)
{
    int x,y;
    x=((struct node*)a)->d;
    y=((struct node*)b)->d;
    if(x!=y)    return x>y?1:-1;
    else    return (struct node*)a-(struct node*)b;
}
int main()
{
    int i,m;
    while(~scanf("%d%d",&len,&m))
    {
        getchar();
        for(i=0;i<m;i++)
        {
            gets(node[i].s);
            node[i].d=cal(node[i].s);
        }
        qsort(node,m,sizeof(node[0]),cmp);
        for(i=0;i<m;i++)    puts(node[i].s);
    }
    return 0;
}