手套_牛客网

链接:https://www.nowcoder.com/questionTerminal/365d5722fff640a0b6684391153e58d8?pos=2&orderByHotValue=0&done=0
来源:牛客网

在地下室里放着n种颜色的手套,手套分左右手,但是每种颜色的左右手手套个数不一定相同。A先生现在要出门,所以他要去地下室选手套。但是昏暗的灯光让他无法分辨手套的颜色,只能分辨出左右手。所以他会多拿一些手套,然后选出一双颜色相同的左右手手套。现在的问题是,他至少要拿多少只手套(左手加右手),才能保证一定能选出一双颜色相同的手套。

给定颜色种数n(1≤n≤13),同时给定两个长度为n的数组left,right,分别代表每种颜色左右手手套的数量。数据保证左右的手套总数均不超过26,且一定存在至少一种合法方案。

测试样例:
  4,[0,7,1,6],[1,5,0,6]
  返回:10(解释:可以左手手套取2只,右手手套取8只)
 
思路:选取一只手的手套,这只手的手套满足:(总数 - 最小 + 1),因为(总数 - 最小 + 1)满足这只手的每种颜色都拿到了,再拿另一只手的任一只手套即可满足题意。
特殊情况:如果被选中手(A)某种颜色的手套数量为0,则在另一只手(B)中应该拿1 + B[A中为0的颜色的手套总和]只手套,因为要确保在B中一定拿到了1只A中的非0颜色的手套。
int findMinimum(int n, vector<int> left, vector<int> right) {
    // write code here
    int sum = 0;
    int leftSum = 0, rightSum = 0;
    int leftMin = INT_MAX, rightMin = INT_MAX;
    for(int i=0;i<n;i++)
    {
        if (left[i]*right[i]==0)
            sum+=(left[i]+right[i]); //sum计算了两只手中有一只某颜色为0时,另一只手该颜色的手套数量的总和
        else
        {
            leftSum += left[i];
            rightSum += right[i];
            leftMin = min(leftMin,left[i]);
            rightMin = min(rightMin,right[i]);
        }
    }
    return sum + min(leftSum-leftMin+1,rightSum-rightMin+1) + 1;
}
 

猜你喜欢

转载自www.cnblogs.com/dabai56/p/10982841.html