美團2018春招編程題第一題 字符串距離 O(n)解法

題目描述：輸入兩個字符串m,n，字符串僅由'a','b'兩種字符組成,且m長度len1大於等於n的長度len2，那麼m有len1-len2+1個長度爲len2的子串，
將這些子串與n進行比較,將字符串距離定義爲：相同位置的元素不一樣則距離加1，如aab與bba的距離爲3

求距離之和

輸入:
aabaa
bba
輸出:
5


思路：n中的每一個字母都要比較len1-len2+1次，如b 須要和 aab比較,b須要和aba比較,a須要和baa比較。
b與aab: 比較結果爲1+1+0 = 2
b與aba：比較結果爲1+0+1 = 2
a與baa：比較結果爲1+0+0 = 1

n中的字母進行比較時，能夠利用到上一個字母比較的結果來簡化步驟

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        String str1 = in.nextLine();
        String str2 = in.nextLine();
        char[] chars1 = str1.toCharArray();
        char[] chars2 = str2.toCharArray();
        int count = 0;//最後要輸出的字符串距離
        int temp = 0;
        for(int j=0;j<str1.length()-str2.length()+1;j++) {//把n中第一個字母比較的結果記錄下來
            if(chars1[j] != chars2[0]) {
                temp++;
            }
        }
        count = count + temp;
        for(int i=1;i<str2.length();i++) {//從n的第二個字母開始，利用前一個字母比較的結果
            if(chars2[i] == chars2[i-1]) {//若是該字母和前一個字母相同
                if(chars2[i-1] != chars1[i-1]) {
                    temp--;
                }
                if(chars2[i] != chars1[i+str1.length()-str2.length()]) {
                    temp++;
                }
            }else {//若是字母和前一個字母不一樣
                if(chars2[i-1] != chars1[i-1]) {
                    temp--;
                }
                temp = str1.length()-str2.length() - temp;
                if(chars2[i] != chars1[i+str1.length()-str2.length()]) {
                    temp++;
                }
            }
            count = count + temp;
        }
        System.out.println(count);
    }
}