AT_abc306_e 题解

一、题目描述：

　　给你一个长度为 $n$ 的序列 $a$ , 你需要进行 $q$ 次操作。

　　　　$将第\ x\ 个元素的值修改为\ v\ 。$

　　每次操作后，输出序列前 $k$ 大的元素和

　　数据范围：$1 \le n,q \le 5 \times 10^5 ,所有数\ -10^9 \le val \le 10^9$

---

 二、解题思路：

　　用两个 $set$ 维护，$s1$ 维护前 $k$ 大，$s2$ 维护非前 $k$ 大 。

　　每次将 $s2$  中最大的与 $s1$ 中最小的进行比较，若能更大就交换。

　　好奇妙的思路啊！不愧是 $jiangly$ ! $orz$ 。时间复杂度 $O(nlog_2^n)$ 。

---

 三、完整代码：

#include<iostream>
#include<set>
#define N 500010
#define ll long long
using namespace std;
multiset <ll> s1,s2;
ll n,k,q,x,y,ans,a[N];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin>>n>>k>>q;
    for(ll i=1;i<=n;i++)
        if(i<=k)    s1.insert(a[i]);
        else        s2.insert(a[i]);
    for(ll i=1;i<=q;i++)
    {
        cin>>x>>y;
        ll t=y-a[x];
        if(k==n)
        {
            a[x]=y;ans+=t;
            cout<<ans<<'\n';
            continue;
        }
        auto iter1=s1.find(a[x]);
        auto iter2=s2.find(a[x]);
        if(iter1!=s1.end())
            s1.erase(iter1),s1.insert(y),ans+=t;
        else    s2.erase(iter2),s2.insert(y);
        a[x]=y;iter1=s1.begin();
        iter2=s2.end(); iter2--;
        x=*iter1,y=*iter2,t=y-x;
        while(t>0)
        {
            ans+=t;
            s1.erase(iter1),s1.insert(y);
            s2.erase(iter2),s2.insert(x);
            iter1=s1.begin();
            iter2=s2.end();iter2--;
            x=*iter1,y=*iter2,t=y-x;
        }
        cout<<ans<<'\n';
    }
    return 0;
}

---

四、写题心得：

　　$set$中位置的访问相对繁琐，但也不是很难。收获经验如下：

　　$1、set\ 好像很难确定某个元素的位置呢。=> Exp++!$

　　$2、访问\ set\ 中最后一个元素可以用\ auto\ iter = s.end() , iter--\ 。=> Exp++$