P6155 修改

P6155 修改

昨晚没睡好，早上状态让人难绷。

容易想到将 $a$ 从大到小排序，依次找到合法的 $a_i^{\prime }$，记修改 $a_i\to a_i^{\prime }$ 为次数为 $p_i$，将 $p_i,b_i$ 逆序相乘即为答案。

考虑如何求解 $a_i^{\prime }$，这是一个类似求 mex 的问题。

$Solution1$

联想到 [CSP-S 2023] 消消乐。可以看下题解的图，很像本题求 mex 的过程，想到可以用并查集求解。

不过我晕乎乎的没敢写。

int find(int x){
	if(f[x]==0)return f[x]=x+1;
	return f[x]=find(f[x]);
}

扫一遍 $a_i$ 即可，由于 $\mathbb{R}(f)\in [1,10^9]$，用 unordered_map 复杂度大概 $O(N{\log }^{2}N)$。

$Solution2$

根据贪心策略，当 $a_i<a_j$ 时，若 $a_i$ 至少要增大到 $a_j$ ，此时 $a_i$ 跨越 $a_j$ 更优。

这个性质表现为包含性，类似括号序列。假如我们从左向右扫，满足后进先出，因此可以用栈模拟过程。
具体地，值域上每个位置 $x$ 可以放一个 $a_i$，因此扫一遍值域，加入 $a_i$ 的同时要求每个 $x$ 上最多只能弹出一个 $a_i$。值域 $10^9$，由于 $n\le 10^5$，意味着 $a_i$ 最多需要 $10^5$ 个位置，因此可以当栈空时手动跳到下一个 $a_i$ 所在位置，复杂度 $O(n)$。

#include<bits/stdc++.h>
#define rep(i,a,b) for(register int i=a;i<=b;i++)
#define dec(i,a,b) for(register int i=a;i>=b;i--)
using namespace std;
typedef unsigned long long ull;
const int N=1e6+5;
int n,a[N],b[N],p0,p[N];
ull ans;
inline int Rd(){
    int s=0,w=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-') w=-1;ch=getchar();}
    while (ch>='0'&&ch<='9') s=(s<<1)+(s<<3)+ch-'0',ch=getchar();
    return s*w;
}
int main(){
    n=Rd();
    rep(i,1,n) a[i]=Rd();rep(i,1,n) b[i]=Rd();
    sort(a+1,a+n+1);sort(b+1,b+n+1,greater<int>());
    stack<int> st;
    int nw=1,x=1;
    auto Push=[&](){while (nw<=n&&a[nw]==x) st.push(a[nw++]);};
    while (1){
        if(st.empty()){
            if(nw<=n) x=a[nw],Push();
            else break;
        }
        else Push();
        p[++p0]=(x++)-st.top();st.pop();
    }
    sort(p+1,p+n+1);
    rep(i,1,n) ans+=1llu*b[i]*p[i];
    printf("%llu",ans);
}