地域划分 编程题

题目：现在有一块长条形的土地，这个土地我们可以看成是由n块小方格连接而成的(这些小方格我们可以将之编号为1到n)。而我们需要将其划分成两个部分，分别种上不同的作物(即作物A和B)，划分必须在某两个小方格之间进行，或者在土地的最左端或最右端，若划分在第i块到第i+1块间进行，则划分后，第1至第i块地种A，剩下的地种B。现在有一些专家对土地进行了检测，他们每个人评估了每块土地适合种的作物。请你找到一个合适的划分，使得其与所有专家的评估最吻合，也就是说，你划分到A而专家评估为B的次数和你划分到B而专家评估为A的次数之和最小。

输入描述:

每组数据给定一个专家评估表land(其中0为评估A，1为评估B)，以及小块数量n(1≤n≤300)，专家评估次数m(1≤m≤300)

输出描述:

请返回你的划分,即i和i+1。若在最左端，则输出0，1；在最右端则输出n,n+1。若有多解输出最靠左的划分。

输入例子:

[[1,1,1,1],[0,0,0,0],[1,0,1,1]],4,3

输出例子:

[0,1]

    vector<int> getPartition(const vector<vector<int> >& land, int n, int m) {
        // write code here
        if(land.empty()) return re;
        vector<int> re;
        int temp=0;       //temp 存放最佳时的i值
        int a=0;          // 存放 0 的个数
        int b=0;          // 存放 1 的个数
        int diff=0;       // 从左到右遍历  a-b  最小的值 
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++){
                if(land[j][i]==0) a++;
                else b++;
            }            
            if(a>b&&a-b>diff) {temp=i+1;diff=a-b;}
        }
        re.push_back(temp);
        re.push_back(temp+1);
        return re;
    }