求数组中两个元素的最小距离

来源:互联网 发布:网络舆情监控资质 编辑:程序博客网 时间:2024/05/22 16:04

一个数组,含有重复元素,给出两个数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;}



0 0
原创粉丝点击