トピックソース
2014ブルーブリッジカップ州
トピックリンク:
https ://www.lanqiao.cn/problems/222/learning//
テストセンター
マージ、逆ペア、ツリー配列
ビデオ説明
https://www.bilibili.com/video/BV1VY411J7nP
アイデア
このトピックは明らかにバブルソートに関連している可能性があります。最初のiiについてはそれを知る必要がありますi人の生徒の左側にいる生徒の数(L i L_iL私は)彼より背が高く、右側に何人のクラスメートがいますか(R i R_iR私は)が彼よりも低い場合、iiの場合はi人の学生が実際に行う交換の数は(L i + R i)(L_i + R_i)(L私は+R私は)、そして前者は実際には逆対であり、後者は順序対であるため、ツリー配列、次に逆の順序対を見つけます。開始は、その合計式で計算できます。詳細については、コードを参照してください。もちろん、マージソートの記述方法は、すべての人の考えに任されています。
コード
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define mod 1000000007
#define endl "\n"
#define PII pair<int,int>
#define INF 0x3f3f3f3f
const int N = 1e6 + 10;
ll a[N],ans[N],n;
struct BIT{
ll tree[N<<2];
ll lowbit(ll x) {
return -x & x;}
void update(ll x,ll k) {
while(x < N) {
tree[x] += k;
x += lowbit(x);
}
}
ll query(ll x) {
ll ans = 0LL;
while(x){
ans += tree[x];
x -= lowbit(x);
}
return ans;
}
void clear(){
memset(tree,0,sizeof tree);
}
}bit;
int main()
{
ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
cin>>n;
for(int i = 1;i <= n; ++i) cin>>a[i],a[i]++;
for(ll i = 1;i <= n; ++i) {
bit.update(a[i],1LL);
ans[i] = i - bit.query(a[i]);
}
bit.clear();
for(ll i = n;i >= 1; --i) {
bit.update(a[i],1LL);
ans[i] += bit.query(a[i] - 1LL);
}
ll res = 0LL;
for(int i = 1;i <= n; ++i)
res += (ans[i] + 1LL) * ans[i] / 2LL;
cout<<res<<endl;
return 0;
}