poj 1007

题目链接

m个字符串以逆序对个数排序后输出

以前都是用线段树写的 现在加上归并

#include<iostream>
#include<string>
#include<algorithm>

using namespace std;

struct xx
{
string str;
int inv;
}a[100];

int ans;

string str;

bool cmp(const xx &a,const xx &b)
{
return a.inv<b.inv;
}

void merge(int l,int r)
{
if(l+1>=r)
{
return;
}
int m=(l+r)>>1;
merge(l,m);
merge(m,r);
for(int i=l;i<m;i++)
{
ans+=lower_bound(str.begin()+m,str.begin()+r,str[i])-(str.begin()+m);
}
inplace_merge(str.begin()+l,str.begin()+m,str.begin()+r);
}

int cal()
{
ans=0;
merge(0,str.size());
return ans;
}

int main()
{
int n,m;
cin>>n>>m;
for(int i=0;i<m;i++)
{
cin>>a[i].str;
str=a[i].str;
a[i].inv=cal();
}
sort(a,a+m,cmp);
for(int i=0;i<m;i++)
cout<<a[i].str<<endl;
system("pause");
return 0;
}
以前的代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>

using namespace std;

#define maxn 4
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

struct xx
{
string str;
int inv;
}a[100];

int sum[maxn<<2];

bool cmp(const xx &a,const xx &b)
{
return a.inv<b.inv;
}

int cal(char c)
{
switch(c)
{
case 'A': return 1;
case 'C': return 2;
case 'G': return 3;
case 'T': return 4;
}
}

int query(int L,int R,int l,int r,int rt)
{
if(L>R)
return 0;
if(L<=l&&r<=R)
{
return sum[rt];
}
int m=(l+r)>>1;
int ret=0;
if(L<=m)
ret+=query(L,R,lson);
if(R>m)
ret+=query(L,R,rson);
return ret;
}

void updata(int c,int l,int r,int rt)
{
if(l==r)
{
if(l==c)
sum[rt]++;
return;
}
int m=(l+r)>>1;
if(c<=m)
updata(c,lson);
else
updata(c,rson);
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

int main()
{
int n,m;
cin>>n>>m;
for(int i=0;i<m;i++)
{
cin>>a[i].str;
memset(sum,0,sizeof(sum));
a[i].inv=0;
for(int j=0;j<a[i].str.size();j++)
{
a[i].inv+=query(cal(a[i].str[j])+1,4,1,4,1);
updata(cal(a[i].str[j]),1,4,1);
}
}
sort(a,a+m,cmp);
for(int i=0;i<m;i++)
cout<<a[i].str<<endl;
// system("pause");
return 0;
}