A. Weird Sum

题目链接 : 

Problem - 1648A - Codeforces

题面 : 

 题意 : 

输入 n m (1≤n*m≤1e5) 和 n 行 m 列的矩阵 a，元素范围 [1,1e5]。
对于矩阵中的所有相同元素对，即满足 a[x1][y1] = a[x2][y2] 的元素对 (a[x1][y1], a[x2][y2])，把 abs(x1-x2) + abs(y1-y2) 加到答案中。
注意 (a,b) 和 (b,a) 只算一次。
输出答案。

思路:

1.结果在数组的横坐标和纵坐标的方向上是可以分离的，不影响；

2.将每一个数字相同的横坐标存进一个vector中，纵坐标一样处理，对于每个单独的vector(p) : 对于第i个数 : 对于p[i]左边有i个数，距离之和为 : (p[i]-p[0])+(p[i]-p[1])+...(p[i]-p[i-1]);
即 i * p[i] - (p[0]+p[1]+p[2]+...+p[i-1]); 

详情请看代码

代码 : 

全放一起 : 

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'

using namespace std;
typedef long long LL;
const int N = 1e5 + 10;

vector<vector<vector<int>>> res(100002, vector<vector<int>>(2, vector<int>()));
LL ans;

// 对于p[i]左边有i个数，距离之和为 : (p[i]-p[0])+(p[i]-p[1])+...(p[i]-p[i-1]);
// 即 i * p[i] - (p[0]+p[1]+p[2]+...+p[i-1]); 
void fac(vector<int>& p){
	LL sum = 0;
	for(int i=0;i<p.size();i++){
		ans += (1LL) * i * p[i] - sum;
		sum += p[i];
	}
}

inline void solve(){
	int n , m , x ; 
	cin >> n >> m;
	for(int i=0;i<n;i++){
		for(int j=0;j<m;j++){
			cin >> x;
			res[x][0].push_back(i); // 横坐标 
			res[x][1].push_back(j); // 纵坐标 
		}
	}
	for(auto& p : res){
		fac(p[0]);
		sort(p[1].begin(),p[1].end());
		fac(p[1]);
	}
	cout << ans << endl;
}
 
int main()
{
    int _ = 1;
    while(_ --) solve();
    return 0;
}

分开处理(性能更优):

#pragma GCC optimize(3)
#include<iostream>
#include<vector>
#include<algorithm>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'


using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
vector<vector<int>> r(N, vector<int>()), c(N, vector<int>());
LL ans;

// 对于p[i]左边有i个数，距离之和为 : (p[i]-p[0])+(p[i]-p[1])+...(p[i]-p[i-1]);
// 即 i * p[i] - (p[0]+p[1]+p[2]+...+p[i-1]); 
LL fac(vector<int>& p) {
	LL sum = 0, ys = 0;
	for (int i = 0; i < p.size(); i++) {
		ys += (1LL) * i * p[i] - sum;
		sum += p[i];
	}
	return ys;
}

inline void solve() {
	int n, m; cin >> n >> m;
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) {
			int x; cin >> x;
			r[x].push_back(i); // 横坐标 
			c[x].push_back(j); // 纵坐标 
		}
	}
	for (auto& p : r) {
		ans += fac(p);
	}
	for (auto& p : c) {
		sort(p.begin(), p.end());
		ans += fac(p);
	}
	cout << ans << endl;
}

int main()
{
	IOS
	int _ = 1;
	while (_--) solve();
	return 0;
}