问题描述:
现在有一块长条形的土地,这个土地我们可以看成是由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]
分析:
1 1 1 1
0 0 0 0
1 0 1 1
该题目实际就是给定一个i,i从0到n,计算一下i左边的1和右边的0 的数目和!!!!
#include <iostream>
#include<vector>
using namespace std;
class Partition {
public:
vector<int> getPartition(const vector<vector<int> >& land, int n, int m) {
int sum = INT_MAX;
int bestSplit=0;
for(int i=-1;i<n;i++)
{//若在最左端,实则是[-1,0]之间。但输出[0,1]
//因此i从-1开始,最后加上1即可
int tmpSum=0;
for(int k=0;k<=i;k++)
{//统计[0,i]列1的个数
for(int j=0;j<m;j++)
if(land[j][k]==1)
tmpSum += 1;
}
for(int k=i+1;k<n;k++)
{//统计[i+1,n-1]列0的个数
for(int j=0;j<m;j++)
if(land[j][k]==0)
tmpSum += 1;
}
if(tmpSum < sum)
{//如果该划分的情况下,sum更小,则更新
sum = tmpSum;
bestSplit=i+1;//i从-1开始,最后加上1即可
}
}
return {bestSplit, bestSplit+1};
}
};
int main()
{
Partition p;
vector<vector<int> > land({{1,1,1,1},{0,0,0,0},{1,0,1,1}});
int n=4, m=3;
vector<int> res = p.getPartition(land, n, m);
for(unsigned int i=0;i<res.size();i++)
cout<<res[i]<<" ";
cout<<endl;
return 0;
}
结果:
0 1
Process returned 0 (0x0) execution time : 0.344 s
Press any key to continue.