洛谷 AT_abc360_c [ABC360C] Move It 题解

题目大意

有 $N$ 个箱子和 $N$ 件物品，编号均为 $1\sim N$。

第 $i$ 件物品放置在第 $A_i$ 个箱子中，重量为 $W_i$。

你可以将一件物品放置到另一个箱子里，而这件物品的重量就是这次操作的代价。

请问要让每个箱子都有意见物品至少要多少代价。

思路

众所周知，想要让代价最少，那么那些能不移动的就都不要移动，所以，如果某个箱子只有一个物品了，那么这个物品就不要移动了。

然后，对于有不止一个物品的箱子，想要达到题目的要求，就要让这个箱子只留下一个物品，剩下的全移走，同时，还要使代价最少。在这个箱子里，要移走的物品数量是不变的，所以要让代价最少，就只能让移走的物品尽可能轻，故要保留这个箱子中最重的一个。

为了方便，可以使用结构体，再进行一边排序，具体的排序方法如下：

• 先按所在箱子编号从小到大排序，若一样，再按重量从小到大排序。

这样就可以使同一个箱子的物品在数组中的位置挨在一起，并且更快找出一个箱子中最重的那一个物品。

Code

#include<iostream>
#include<algorithm>
using namespace std;
struct stru{//结构体
	int w,a;
}arr[100000];
bool cmp(stru x,stru y) {//sort排序自定义规则
	if(x.a!=y.a) return x.a<y.a;
	return x.w<y.w;
}
signed main() {
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	int n,ans=0;
	cin>>n;
	for(int i=0;i<n;++i) cin>>arr[i].a;
	for(int i=0;i<n;++i) cin>>arr[i].w;
	sort(arr,arr+n,cmp);//排序
	for(int i=0;i<n;++i) if(i<n-1/*防止数组出界*/&&arr[i].a==arr[i+1].a/*如果下一个也是这个箱子说明这个不是最重的物品*/) ans+=arr[i].w;//累加答案
	cout<<ans;
	return 0;
}