字符的最短距离

flag

软件学院大三党，每天一道算法题，第28天

题目介绍

给定一个字符串 S 和一个字符 C。返回一个代表字符串 S 中每个字符到字符串 S 中的字符 C 的最短距离的数组。

思路

仔细观察输出数组，发现是一个有峰值的序列，我们首先要找到所有0的位置，然后从左到右遍历一次，找到每个值离左面最近0的距离，再从右到左遍历，找到每个值里右面最近0的距离，如果更小就覆盖之前的值

关键代码

    public static int[] shortestToChar(String S, char C){
        char []arr=S.toCharArray();
        int len=arr.length;
        int []result=new int[len];//保存结果
        for (int i=0;i<len;i++){//初始化，将C对应的结果设为0，其它设为max_Value
            if(arr[i]==C)
                result[i]=0;
            else
                result[i]=Integer.MAX_VALUE;
        }
        int count=-1;//计数器，记录两个C直接的距离，不能初始化为0，因为arr[0]可能为C
        for(int i=0;i<len;i++){//从左到右遍历一次
            if(result[i]==0){
                count=i;
            }
            else if(result[i]!=0&&count!=-1) {
                result[i]=i-count;
            }
        }
        count=0;//初始化

        for(int i=len-1;i>=0;i--){//从右往左遍历
            if(result[i]==0){
                count=i;
            }
            else if(result[i]!=0&&count!=0&&result[i]>count-i)
                result[i]=count-i;
        }
        return result;
    }