Java学习手册:(数据结构与算法-数组)如何求数组中两个元素的最小距离?

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: https://blog.csdn.net/MaybeForever/article/details/101351330

问题描述:

给定一个数组,数组中含有重复元素,给出两个数n1和n2,求这两个数字在数组中所出现位置的最小距离,例如,数组中,4和8的最小的距离为2。

方法一:

遍历数组,会遇到以下两种情况:

(1)当遇到n1时,记录下n1值对应的数组下标的位置n1_index,通过求n1_index与上次遍历到n2的下标值n2_index的差,可以求出最近一次遍历到n1与n2的距离。

(2)当遇到n2时,记录下n2值对应的数组下标的位置n2_index,通过求n2_index与上次遍历到n1的下标值n1_index的差,可以求出最近一次遍历到n1与n2的距离。

定义一个遍历min_dist记录n1与n2的最小距离,在以上两种情况下,每次求出n1与n2的距离后与min_dist比较,求出最小值。

只需对数组进行一次遍历即可求出最小距离,时间复杂度为O(n)。

方法一代码如下:

package com.haobi;

public class Test23 {
	public static void main(String[] args) {
		int[] a = {4,5,6,4,7,4,6,4,7,8,5,6,4,3,10,8};
		System.out.println(minDistance(a, 4, 8));
	}
	
	public static int minDistance(int[] a,int n1,int n2) {
		if(a==null) {
			return Integer.MAX_VALUE;
		}
		int len = a.length;
		int n1_index = -1;
		int n2_index = -1;
		int min_dist = Integer.MAX_VALUE;
		for(int i=0;i<len;i++) {
			if(a[i] == n1) {
				n1_index = i;
				if(n2_index >= 0) {
					min_dist = min(Math.abs(min_dist), Math.abs(n1_index - n2_index));
				}
			}
			if(a[i] == n2) {
				n2_index = i;
				if(n1_index >= 0) {
					min_dist = min(Math.abs(min_dist), Math.abs(n1_index - n2_index));
				}
			}
		}
		return min_dist;	
	}
	
	public static int min(int a, int b) {
		return a>b ? b: a;
	}
	
}

程序输出结果如下:

2

猜你喜欢

转载自blog.csdn.net/MaybeForever/article/details/101351330