版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_40924940/article/details/84948705
这题有毒,结果printf已知WA 改成 cout 就对了。。。。很毒。。。
给一个打乱顺序的 n 个数排序,每次交换两个数花费是 X+Y 。
还是树状数组,但是开两个数组,一个统计数字个数,一个统计数字大小。
对于当前的数 x 我们可以知道他前边有几个大的数,她也需要进行几次交换,每次交换也需要加上那些被交换的数的大小,那么结果显而易见,就出来了。
以下是AC代码
扫描二维码关注公众号,回复:
4491785 查看本文章
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define ll long long
const int INF = 0x3f3f3f;
const int maxn = 1e6 + 5 ;
const ll lowbit(ll x){ return x&-x; }
ll arr[maxn];
ll num[maxn];
ll val[maxn];
ll n;
void init()
{
memset(arr,0,sizeof(arr));
memset(num,0,sizeof(num));
memset(val,0,sizeof(val));
}
ll getnum(ll x)
{
ll ans = 0;
for(ll i = x;i > 0;i -= lowbit(i))
ans += num[i];
return ans;
}
ll getval(ll x)
{
if(x == 0)
return 0;
ll ans = 0;
for(ll i = x;i > 0;i -= lowbit(i))
ans += val[i];
return ans;
}
void add(ll x,ll v)
{
for(ll i = x;i < maxn;i += lowbit(i))
{
num[i] += 1;
val[i] += v;
}
}
int main()
{
while(~scanf("%d",&n))
{
init();
ll sum = 0;
ll ans = 0;
for(int i = 1;i <= n;i++)
{
scanf("%d",&arr[i]);
add(arr[i],arr[i]);
sum += arr[i];
ans += arr[i] * (i - getnum(arr[i]));
ans += sum - getval(arr[i]);
}
cout<<ans<<endl;
}
return 0;
}