LeetCode 1573. 分割字符串的方案数

题目链接：LeetCode 1573. 分割字符串的方案数

题意：

       给你一个二进制串 s  （一个只包含 0 和 1 的字符串），我们可以将 s 分割成 3 个 非空 字符串 s1, s2, s3 （s1 + s2 + s3 = s）。

请你返回分割 s 的方案数，满足 s1，s2 和 s3 中字符 '1' 的数目相同。

由于答案可能很大，请将它对 10^9 + 7 取余后返回。

示例 1：

输入：s = "10101"
输出：4
解释：总共有 4 种方法将 s 分割成含有 '1' 数目相同的三个子字符串。
"1|010|1"
"1|01|01"
"10|10|1"
"10|1|01"

解题思路：

  由于PHP 不定义数据范围，所以，不存在强制类型转化，其他语言需要注意

   最中间的1左右两边 0 的个数   各加1相乘

如果全为0，那么答案为  0的个数减去2的等差数列

class Solution {

    /**
     * @param String $s
     * @return Integer
     */
    function numWays($s) {
    	$mod = 1e9+7;
    	$len = strlen($s);
    	$k = 0;
    	for($i = 0; $i < $len; $i++) {
    		if($s[$i] == "1") {
    			$k++;
    		}
    	}
    	if($k % 3 != 0) {      //  1 的个数不能整除3，返回0
    		return 0;
    	} else {
    		$k1 = 0; $k2 = 0; $k3 = 0;
    		$a1 = 0; $a2 = 0;
    		if($k == 0) {
    			return intval((1+$len-2)*($len-2)/2) % $mod;  // 1 - len 的等差数列
    		}
    		for($i = 0; $i < $len; $i++) {
    			if($s[$i] == "1") {
    				$k1++;
    			}
    			if($k1 == $k / 3) {
    				$k1 = $i; break;
    			}
    		}
    		$k1++;
    		for($i = $k1; $i < $len; $i++) {
    			while($i < $len && $s[$i] == "0") {
    				$i++;
    				$a1++;         // 左边0的个数
    			}
    			while($i < $len && $k2 != $k / 3) {
    				if($s[$i] == "1") {
    					$k2++;
    				}
    				$i++;
    			}
    			while($s[$i] == "0") {
    				$a2++;     // 右边0的个数
    				$i++;
    			}
    			break;
    		}
    		return ($a1+1) * ($a2+1) % $mod;
    	}
    }
}