题解:Minimize Inversions Number
前言
好题。
思路分析
分析一下答案的组成:
令 , 表示选出的集合, 表示集合 的逆序对数, 表示这个序列的逆序对数:
不难发现问题在于如何决策 这一项。
做一点观察:对于任意两个 满足 ,都满足 优先于 被选择。
考虑用调整法进行证明,将 画到二维平面上,分析这两个点划分出的九个象限,发现如果我们将 点向右下方移动,那么序列的逆序对数会变大。
因此,我们总可以找到一个更合适的 ,使得答案之和最小。
同时,因为调整之后的下标会增加,调整都会在有限步数内结束。
这样我们就证明了结论的正确性。
设 ,那么不难发现:
所以答案变为:
我们惊喜的发现这个式子对于 是独立的!
具体地,我们按 排序,每次选择前 大即可。
用树状数组求逆序对即可。
总体复杂度 。
代码实现
#include<bits/stdc++.h>
#define int long long
using namespace std;
int t,n,ans,a[500005],b[500005];
#define lowbit(i) (i&(-i))
int c[500005];
void modify(int x,int k){
for(int i=x;i<=n;i+=lowbit(i)){
c[i]+=k;
}
}
int query(int x){
int ans=0;
for(int i=x;i;i-=lowbit(i)){
ans+=c[i];
}
return ans;
}
bool cmp(int x,int y){
return x>y;
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>t;
while(t--){
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
b[i]=i-2*a[i]+1;
ans+=query(n)-query(a[i]);
modify(a[i],1);
}
sort(b+1,b+1+n,cmp);
cout<<ans<<' ';
for(int i=1;i<=n;i++){
ans-=b[i];
cout<<ans-i*(i-1)/2<<' ';
}
cout<<'\n';
for(int i=1;i<=n;i++){
c[i]=0;
}
ans=0;
}
return 0;
}
本文作者:Kenma
本文链接:https://www.cnblogs.com/Kenma/p/18698551
版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步