思路
题意就不赘述了,对于一个数字插入deque前面的话对答案的贡献就是当前deque中比该数字小的个数,一个数字插入到deque的后面的话对答案的贡献就是当前deque中比该数字大的个数,和上一个数字插到前面和后面无关。所以这里比较容易得出思路,对于a数组中第i位上的数字,计算它对答案的贡献需要算出:1.前面的数字比他小的个数2.他前面的数字比他大的个数,然后取小的那个个数就是第i位对答案的贡献。对每一位都求一次然后加起来就行了。
代码示例
TT 先看一下蠢比做法(错误做法)
distance函数对于随机访问迭代器的复杂度为O(1),对于非随机访问迭代器的复杂度位O(N)不是O(1),写的时候忘记了这点,一直tle。我太蠢了555 /(ㄒoㄒ)/~~
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int T;
int n;
int a[200005];
void solve()
{
ll ans=0;
multiset<int>s;//用multiset存储数字
multiset<int>::iterator lower;//存储小于a[i]的第一个数字的位置
multiset<int>::iterator bigger;//存储大于a[i]的第一个数字位置
for(int i=1;i<=n;i++)
{
if(!s.size())//判断set是否为空
{
s.insert(a[i]);//set中加入a[i]
continue;
}
int low,big;
lower=s.lower_bound(a[i]);//查找小于等于a[i]的迭代器,lower-1就是比他小的位置了
bigger=s.upper_bound(a[i]);//查找大于a[i]的迭代器
big=s.size()-distance(s.begin(),bigger);//总长度长度减去一开始到大于a[i]的长度就是大于a[i]的个数
low=distance(s.begin(),lower);//开始到lower的长度-1就是比他a[i]小的数字个数
if(lower==s.begin())
low=0;
if(bigger==s.end())
big=0;
ans+=min(big,low);//更新答案
s.insert(a[i]);//set中加入a[i]
}
printf("%lld\n",ans);
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
solve();
}
return 0;
}
正确做法
对每个数字求小于它的个数和大于它的个数,先对原来的数组进行去重,然后计算出每个a[i]在去重的数组中的排名p[i],p[i]=1说明第i个数字是去重数组中最小的数字,p[i]=2说明第i个数字是去重数组中第二小的数字。令query(1,p[i])表示求第i位之前排名小于等于p[i]的数字的个数,令update(p[i])表示排名为p[i]的数字个数+1,比较容易就能看出是线段树的区间查询+单点修改。具体看代码喽=3=
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int T;
int n;
vector<int>v;
ll s[1000000];//区间和
ll a[200010];
void pushup(int rt)
{
s[rt]=s[rt<<1]+s[rt<<1|1];
}
void update(int L,int R,int rt,int x)//单点更新
{
if(L==R)
{
s[rt]++;
return;
}
int mid=L+R>>1;
if(x<=mid)update(L,mid,rt<<1,x);
if(x>mid)update(mid+1,R,rt<<1|1,x);
pushup(rt);
}
ll query(int L,int R,int l,int r,int rt)//区间查询
{
if(r<l)
return 0;
if(l<=L&&r>=R)
return s[rt];
int mid=L+R>>1;
ll ans=0;
if(l<=mid)ans+=query(L,mid,l,r,rt<<1);
if(r>mid)ans+=query(mid+1,R,l,r,rt<<1|1);
return ans;
}
int main()
{
scanf("%d",&T);
while(T--)
{
ll ans=0;
scanf("%d",&n);
memset(s,0,sizeof(s));
v.clear();
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]),v.push_back(a[i]);
sort(v.begin(),v.end());
v.erase(unique(v.begin(),v.end()),v.end());//去重
for(int i=1;i<=n;i++)//求出去重后的从小到大排名p[i]并赋值给a[i]
a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1;
for(int i=1;i<=n;i++)
{
ll low=query(1,n,1,a[i]-1,1);//求(1,a[i]-1)的个数,也就是比第i个数字小的个数
ll big=i-1-query(1,n,1,a[i],1);//求比第i个数字大的个数
update(1,n,1,a[i]);//第a[i]小的数字+1
ans+=min(low,big);//更新答案
}
printf("%lld\n",ans);
}
return 0;
}
树状数组同理哦
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int T;
int n;
ll s[200010];//区间和
ll a[200010];
vector<ll>v;
int lowbit(int x)
{
return x&-x;
}
ll sum(int x)//区间查询
{
ll ans=0;
for(int i=x;i;i-=lowbit(i))
ans+=s[i];
return ans;
}
int add(int x,int val)//单点更新
{
for(int i=x;i<=n;i+=lowbit(i))
s[i]+=val;
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
ll ans=0;
memset(s,0,sizeof(s));
v.clear();
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
v.push_back(a[i]);
}
sort(v.begin(),v.end());
v.erase(unique(v.begin(),v.end()),v.end());//去重
for(int i=1;i<=n;i++)//求出去重后从小到大的排名p[i]并赋值给a[i]
a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1;
for(int i=1;i<=n;i++)
{
ll low=sum(a[i]-1);//排名小于等于a[i]-1的个数
ll big=i-1-sum(a[i]);//计算i-1-当前排名小于等a[i]的个数
add(a[i],1);
ans+=min(low,big);//更新答案
}
printf("%lld\n",ans);
}
return 0;
}