Codeforces Round 775(Div.2) Problem C Weird Sum(匿名函数的使用)

原题链接

C++ Lambda函数（匿名函数）的使用方法

​ 首先了解一下 C++ 匿名函数的基本使用方法

基本语法

//[捕获列表](参数列表)->返回类型{函数体}
auto f = [](int a, int b)->int {
	return a + b;
};

捕获列表

​ []：空捕获列表，Lambda不能使用所在函数中的变量。

​ [&]：隐式捕获列表，Lambda体内使用的局部变量都按引用方式传递。

​ [=]：隐式捕获列表，Lambda体内使用的局部变量都按值传递。

返回类型

​ 可以不填，默认为return语句返回值的返回类型。

应用

​ $sort$排序

	auto cmp = [&](pair<int, int> p1, pair<int, int> p2){
		return p1.second < p2.second;
	};
	sort(v.begin(), v.end(), cmp);

	sort(v.begin(), v.end(), [&](auto a, auto b){
		return a.second < b.second;
	});

本题题意

​ 给一个$n\ast m$的表格，每个单元格里面有一个数字，代表一种颜色，求所有两两相同颜色的单元格的曼哈顿距离之和。

思路

​ 1.用$map<int,pair<int,int>>$记录所有颜色所在的单元格。

​ 2.对某一种颜色，我们将其横坐标排序，遍历每一个横坐标，假设当前位置为$i$，我们记录前面所有的横坐标总个数$cnt$以及所有横坐标之和$sum$，不难证明，当前横坐标和前面所有横坐标的距离之和为$cnt\ast i-sum$，因此，我们可以得到当前颜色的所有横坐标两两组合的距离之和，并且由于我们是从小到大枚举，所以得到的答案就是两两横坐标距离差的绝对值之和，即曼哈顿距离的要求。

​ 3.同样的，我们对纵坐标进行操作，最后两者相加即是答案。

代码

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
int main()
{
	int n, m;
	map<int, vector<pair<int, int>>> mp;

	cin >> n >> m;
	for(int i = 0; i < n; i ++)
		for(int j = 0; j < m; j ++)
		{
			int x;
			cin >> x;
			mp[x].push_back({i, j});
		}

	// auto cmp = [&](pair<int, int> p1, pair<int, int> p2){
	// 	return p1.second < p2.second;
	// };

	auto sol = [&](vector<pair<int, int>> v){
		long long res = 0;
		sort(v.begin(), v.end());
		long long cnt = 0, sum = 0;
		for(auto &[i, j] : v)
		{
			res += i * cnt - sum;
			sum += i;
			cnt ++;
		}
		cnt = sum = 0;
		// sort(v.begin(), v.end(), cmp);

		sort(v.begin(), v.end(), [&](auto a, auto b){
			return a.second < b.second;
		});

		for(auto &[j, i] : v)
		{
			res += i * cnt - sum;
			sum += i;
			cnt ++;
		}
		return res;
	};

	long long ans = 0;
	for(auto &[c, v] : mp)
		ans += sol(v);

	cout << ans << endl;
	return 0;
}