1579E2 Array Optimization by Deque

思路

题意就不赘述了，对于一个数字插入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;
}