题目描述:
现在有一块长条形的土地,这个土地我们可以看成是由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 }
分析:
根据题目要求,我们需要找到一个边界,这个边界的左边都为 0 ,右边都为 1 ,而且,这个划分必须与专家评估的结果不同的土地(也就是方格)数最少,
如下示例:
输入:
[[1,1,1,1],[0,0,0,0],[1,0,1,1]],4,3
也就是 4 块土地,3次评估
则所有的情况如下图:
将每种情况对应的专家评估不吻合的土地数加起来:
遇到不吻合的土地数相同的情况下(如上面 1111 和 0011),取最左的划分,也就是 1111,则划分为 0,1
再举一个例子:
所有情况如下:
所以划分就为: 1 , 2
根据上述思想,写代码就容易多了:
vector<int> getPartition(const vector<vector<int> >& land, int n, int m) { int* a = new int[n]; for (int i = 0; i < n; ++i) //全部置1,从全 1 的情况开始 { a[i] = 1; } int left = 0; //划分的左边界 int right = 1; //计划分的右边界 int mindif = n*m; //最少的不吻合的土地数 int difCount = 0; //用来记录不吻合的个数 int zeroCount = 0; //划分序列中 0 的个数 int dif = 0; //每种情况的不吻合的土地数 for (int index = 0; index < n + 1; ++index) //一共有n+1种可能组合 { for (int i = 0; i < m; ++i) { for (int j = 0; j < n; ++j) { if (land[i][j] != a[j]) ++difCount; } dif += difCount; difCount = 0; } if (dif <= mindif) { if (dif < mindif) { mindif = dif; left = zeroCount; right = zeroCount + 1; } else if (zeroCount < left) { left = zeroCount; right = zeroCount + 1; } } dif = 0; if (zeroCount < n) a[zeroCount++] = 0; } delete[] a; a = NULL; vector<int> ret; ret.push_back(left); ret.push_back(right); return ret; }
时间: 2024-10-16 18:10:54