一个数组,含有重复元素,给出两个数num1和num2,求这两个数字在数组中出现的位置的最小距离。
要求O(n)时间复杂度,O(1)空间复杂度。
思路:设置2个指针,数组扫描一遍,出现num1和num2,分别设置相应的指针变量,然后计算最小距离。
代码如下:
#include <stdio.h>
int min(int a, int b){
return a<b?a:b;
}
int findMinDist(int a[], int len, int num1, int num2){
if ( a==NULL || len<2 ) {
return 0;
}
int p1 = -1;
int p2 = -1;
int minDist = len;
for (int i=0; i<len; i++) {
if (a[i]==num1) {
p1 = i;
if (p2!=-1) {
minDist = min(minDist, p1-p2);
}
}
if (a[i]==num2) {
p2 = i;
if (p1!=-1) {
minDist = min(minDist, p2-p1);
}
}
}
return minDist;
}
int main(int argc, const char * argv[])
{
int a[]={2,4,3,5,5,2,1,3,6,7,2,2,3,4,4};
int len = 15;
int num1 = 3;
int num2 = 2;
printf("MinDist = %d\n",findMinDist(a, len, num1, num2));
return 0;
}