P11001 [蓝桥杯 2024 省 Python B] 连连看
题目描述
小蓝正在和朋友们玩一种新的连连看游戏。在一个 n × m n \times m n×m 的矩形网格中,每个格子中都有一个整数,第 i i i 行第 j j j 列上的整数为 A i , j A_{i,j} Ai,j。玩家需要在这个网格中寻找一对格子 ( a , b ) (a,b) (a,b) 和 ( c , d ) (c,d) (c,d) 使得这两个格子中的整数 A a , b A_{a,b} Aa,b 和 A c , d A_{c,d} Ac,d 相等,且它们的位置满足 ∣ a − c ∣ = ∣ b − d ∣ > 0 |a-c| = |b-d| > 0 ∣a−c∣=∣b−d∣>0。请问在这个 n × m n \times m n×m 的矩形网格中有多少对这样的格子满足条件。
输入格式
输入的第一行包含两个正整数 n n n 和 m m m,用一个空格分隔。
接下来是 n n n 行,第 i i i 行包含 m m m 个正整数 A i , 1 , A i , 2 , … , A i , m A_{i,1}, A_{i,2}, \dots, A_{i,m} Ai,1,Ai,2,…,Ai,m,相邻整数之间用一个空格分隔。
输出格式
输出一行包含一个整数表示答案。
输入输出样例 #1
输入 #1
3 2
1 2
2 3
3 2
输出 #1
6
说明/提示
一共有以下 6 对格子: ( 1 , 2 ) − ( 2 , 1 ) (1,2)−(2,1) (1,2)−(2,1), ( 2 , 2 ) − ( 3 , 1 ) (2,2)−(3,1) (2,2)−(3,1), ( 2 , 1 ) − ( 3 , 2 ) (2,1)−(3,2) (2,1)−(3,2), ( 2 , 1 ) − ( 1 , 2 ) (2,1)−(1,2) (2,1)−(1,2), ( 3 , 1 ) − ( 2 , 2 ) (3,1)−(2,2) (3,1)−(2,2), ( 3 , 2 ) − ( 2 , 1 ) (3,2)−(2,1) (3,2)−(2,1)。
数据范围
对于 20 % 20\% 20% 的评测用例, 1 ≤ n , m ≤ 50 1 \leq n, m \leq 50 1≤n,m≤50;
对于所有评测用例, 1 ≤ n , m ≤ 1000 1 \leq n, m \leq 1000 1≤n,m≤1000, 1 ≤ A i , j ≤ 1000 1 \leq A_{i,j} \leq 1000 1≤Ai,j≤1000。
C++实现
#include <bits/stdc++.h>
using namespace std;
map <pair<int,int>,int> mp1,mp2;
int main(){
int n,m,ans=0;
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
int a;
cin>>a;
ans+=mp1[make_pair(a,i-j)]++,ans+=mp2[make_pair(a,i+j)]++;
}
cout<<ans*2;//乘2输出
return 0;
}
后续
接下来我会不断用C++来实现信奥比赛中的算法题、GESP考级编程题实现、白名单赛事考题实现,记录日常的编程生活、比赛心得,感兴趣的请关注,我后续将继续分享相关内容