统计负数 --- lintcode832

点击打开原题

已知：

给定一个二维数组，二维数组从左往右是排序的，从上往下也是排序的，求其中负数个数。

思路：

考虑到减少时间复杂度，我们把每次找到的最后一个负数的纵轴下标作为下一次遍历的最大值（因为根据这个数组的排序方式，负数肯定集中在左上，且正负数切割线是 / 形状）

public class Solution {
    /**
     * @param nums: the sorted matrix
     * @return: the number of Negative Number
     */
    public int countNumber(int[][] nums) {
        
       int ret = 0;

		int index = nums[0].length - 1;
		for (int i = 0; i < nums.length; i++) {
			if (nums[i][0] > 0) {
				break;
			}

			for (int j = index; j >= 0; j--) {
				if (nums[i][j] < 0) {
					ret += j + 1;
					index = j;
					break;
				}
			}
		}

		return ret;
    }
}