進擊的小白Day036——LeetCode刷題筆記016：實現strStr()

題目來自LeetCode。
題目描述:
實現 strStr() 函數。
給定一個 haystack 字符串和一個 needle 字符串，在 haystack 字符串中找出 needle 字符串出現的第一個位置 (從0開始)。如果不存在，則返回 -1。
示例 1:
輸入: haystack = “hello”, needle = “ll”
輸出: 2
示例 2:
輸入: haystack = “aaaaa”, needle = “bba”
輸出: -1

貼代碼：

int strStr(char* haystack, char* needle) {
    int i = 0, j = 0, n1 = 0, n2 =0, flag = -1;
    
    n1 = strlen(haystack);
    n2 = strlen(needle);
    if ((n1 == 0 && n2 == 0) || (n1 != 0 && n2 == 0))
    {
        return 0;
    }
    else if (n1 < n2)
    {
        return -1;
    }
    for (i = 0; i < (n1 - n2 + 1); i++)
    {
        if (haystack[i] == needle[0])
        {
            flag = i;
            for (j = 1; j < n2; j++)
            {
                if (haystack[i + j] != needle[j])
                {
                    flag = -1;
                    break;
                }
            }
            if (j == n2)
            {
                break;
            }
        }
    }
    
    return flag;
}