有n个物品,每个物品有k个属性,第i件物品的第j个属性用一个正整数标记为Ai,j,两个不同的物品i,j被称为完美对当且仅当Ai,1+Aj,1=Ai,2+Aj,2=…=Ai,k+Aj,k,求完美对的个数。
分析
常规做法自然是物品之间两两比较,这样肯定不会通过的,复杂度太高。
对上式进行变形,把一个物品的属性放到一边。
Ai,1-Ai,2=-(Aj,1-Aj,2)
Ai,1-Ai,3=-(Aj,1-Aj,3)
.
.
.
Ai,2-Ai,3=-(Aj,2-Aj,3)
.
.
.
抽象出来就是对于一个物品的属性,将相邻的两个属性相减之后,完美对的两个物品,属性相减之后的值互为相反数。
等于是对每个物品的属性值进行序列化,完美对的两个物品对应的序列中每个数字都互为相反数。
import java.util.*;
import java.io.*;
public class Main{
public static void main(String[] args)throws IOException{
BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
String[] strs=br.readLine().split(" ");
int n=Integer.parseInt(strs[0]);
int k=Integer.parseInt(strs[1]);
int[][] num=new int[n][k];
for(int i=0;i<n;i++){
strs=br.readLine().split(" ");
for(int j=0;j<k;j++){
num[i][j]=Integer.parseInt(strs[j]);
}
}
Map<String,Integer> map=new HashMap<>();
int ans=0;
for(int i=0;i<n;i++){
StringBuilder str1=new StringBuilder();
StringBuilder str2=new StringBuilder();
for(int j=0;j<k-1;j++){
str1.append(num[i][j]-num[i][j+1]);
str2.append(num[i][j+1]-num[i][j]);
}
ans+=map.getOrDefault(str1.toString(),0);
map.put(str2.toString(),map.getOrDefault((str2.toString()),0)+1);
}
System.out.println(ans);
}
}
因篇幅问题不能全部显示,请点此查看更多更全内容