搜索
您的当前位置:首页正文

算法学习笔记-完美对

来源:好走旅游网

有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);
        
        
    }
}

因篇幅问题不能全部显示,请点此查看更多更全内容

Top