djfuuxjz @ 2019-06-16 10:08:07
权值线段树只得了40分:(
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
struct in
{
int w;
int k;
}a[500005];
struct node
{
int l,r,w;
}tr[4000005];
void build(int l,int r,int ro)
{
tr[ro].l = l,tr[ro].r = r;
if(l == r)tr[ro].w = 0;
else
{
int mid = (l + r) / 2;
build(l,mid,2 * ro);
build(mid + 1,r,2 * ro + 1);
tr[ro].w = tr[2 * ro].w + tr[2 * ro + 1].w;
}
}
int find(int l,int r,int ro)
{
if(tr[ro].l == l && tr[ro].r == r)return tr[ro].w;
else
{
int mid = (tr[ro].l + tr[ro].r) / 2;
if(r <= mid)return find(l,r,2 * ro);
else if(l > mid)return find(l,r,2 * ro + 1);
else return find(l,mid,2 * ro) + find(mid + 1,r,2 * ro + 1);
}
}
void add(int x,int ro)
{
if(tr[ro].l == tr[ro].r && tr[ro].l == x)tr[ro].w ++;
else
{
int mid = (tr[ro].l + tr[ro].r) / 2;
if(x <= mid)add(x,2 * ro);
else if(x > mid)add(x,2 * ro + 1);
tr[ro].w = tr[2 * ro].w + tr[2 * ro + 1].w;
}
}
bool cmp(in x,in y)
{
return x.w < y.w;
}
bool cmpp(in x,in y)
{
return x.k < y.k;
}
int main()
{
int n;
int ans = 0;
scanf("%d",&n);
for(int i = 1;i <= n;i ++)
{
scanf("%d",&a[i].w);
a[i].k = i;
}
sort(a + 1,a + 1 + n,cmp);
for(int i = 1;i <= n;i ++)
{
a[i].w = i;
}
sort(a + 1,a + 1 + n,cmpp);
build(1,n,1);
for(int i = 1;i <= n;i ++)
{
ans += find(a[i].w,n,1);
add(a[i].w,1);
}
printf("%d",ans);
return 0;
}
by charliegong @ 2019-06-16 11:40:20
只会写归并,没学线段树