[NERC2018] Lazyland（排序&贪心）

题目背景

翻译自 NERC 2018 L 题。

题目描述

一个城市里有 $n$ 个人，和 $k$ 种职业，其中，每个人都有一个现在正在从事的职业 $a_i$，但是很遗憾，由于某些职业的空缺，使得这个城市根本无法继续正常生存下去。

你作为一城之主，自然不希望看到自己的城市这样子下去，你决定说服其中的某些人，使得每种职业都有人在做，对于每个人 $i$，你需要耗费 $b_i$ 的时间去说服。

你需要求出你去说服的时间的最小值。

输入格式

第一行两个整数 $n$ 和 $k(1\le k\le n\le 10^5)$，分别表示 $n$ 个人和 $k$ 种职业。

第二行 $n$ 个整数 $a_i(1\le a_i\le k)$，表示第 $i$ 个人正在从事的职业 。

第三行 $n$ 个整数 $b_i(1\le b_i\le 10^9)$，表示第 $i$ 个人被说服去做别的职业的时间。

输出格式

输出去说服市民的最小时间。

样例 #1

样例输入 #1

8 7
1 1 3 1 5 3 7 1
5 7 4 8 1 3 5 2

样例输出 #1

10

样例 #2

样例输入 #2

3 3
3 1 2
5 3 4

样例输出 #2

0

提示

对于所有的数据，保证 $1\le k\le n\le 10^5$，$1\le a_i\le k$，$1\le b_i\le 10^9$。

对于样例一，分别令 $1$，$6$，$8$ 号市民去从事 $2$，$4$，$6$ 号职业。

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define inf 1e18
const int mod=1e9+7;
const int N=2e5+5;
int n,m,vis[N],a[N];
vector<pair<int,int>>e[N];
void solve(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		vis[a[i]]=1;
	}
	for(int i=1;i<=n;i++){
		int x;
		cin>>x;
		e[a[i]].push_back({i,x});
	}
	vector<int>v;
	for(int i=1;i<=m;i++){
		if(e[i].size()==0) continue;
		int id,t=0;
		for(auto it:e[i]){
			if(it.second>t){
				id=it.first;
				t=it.second;
			}
		}
		for(auto it:e[i]){
			if(it.first!=id){
				v.push_back(it.second);
			}
		}
	}
	int ans=0,f=0;
	sort(v.begin(),v.end());
	for(int i=1;i<=m;i++){
		if(!vis[i]){
			ans+=v[f++];
		}
	}
	cout<<ans;
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	int tt=1;
//	cin>>tt;
	while(tt--) solve();
	return 0;
}