题目描述
在地下室里放着n种颜色的手套,手套分左右手,但是每种颜色的左右手手套个数不一定相同。A先生现在要出门,所以他要去地下室选手套。但是昏暗的灯光让他无法分辨手套的颜色,只能分辨出左右手。所以他会多拿一些手套,然后选出一双颜色相同的左右手手套。现在的问题是,他至少要拿多少只手套(左手加右手),才能保证一定能选出一双颜色相同的手套。
给定颜色种数n(1≤n≤13),同时给定两个长度为n的数组left,right,分别代表每种颜色左右手手套的数量。数据保证左右的手套总数均不超过26,且一定存在至少一种合法方案。
测试样例:
4,[0,7,1,6],[1,5,0,6]
返回:10(解释:可以左手手套取2只,右手手套取8只)
class Gloves {
public:
//对于非0递增序列a1,a2...an,要想最终取值覆盖每一个种类
//n = sum(a1...an) - a1 + 1(也就是总数减去最小值之后加一)
//所以对于左右手手套颜色都有数量的序列,取左边和右边的最小值即可
int findMinimum(int n, vector<int> left, vector<int> right)
{
int left_sum = 0, left_min = INT_MAX;
int right_sum = 0, right_min = INT_MAX;
int sum = 0;
//遍历每一种颜色的左右手套序列
for(int i=0;i<n;i++)
{
//对于有0存在的颜色手套,累加
if(left[i]*right[i]==0)
sum += left[i] + right[i];
//对于左右手都有的颜色手套,执行上面的策略
//找到最小值和总数
else
{
left_sum += left[i];
right_sum += right[i];
left_min = min(left_min, left[i]);
right_min = min(right_min, right[i]);
}
}
//结果为有左右都有数量的手套序列的结果+有0存在的手套数+最后再加一肯定就能保证了
return sum + min(left_sum-left_min+1, right_sum-right_min+1) + 1;
}
};