C++ 牛群 字符串类型简单统计

约翰有 n
头奶牛。

每头奶牛的品种由一个小写字母表示,不同奶牛的品种可能不同。

请问,能否将这些奶牛分为两个非空的牛群,并使得每个牛群都恰好包含两种奶牛。

例如,如果一共有 5
头奶牛,品种依次为 aabcc,则可将它们分为两个牛群 ab 和 acc,其中,第 1
个牛群恰好包含 a、b 两个品种的奶牛,第 2
个牛群恰好包含 a、c 两个品种的奶牛。

输入格式
共一行,包含一个长度为 n
的由小写字母构成的字符串,其中第 i
个字符表示第 i
头奶牛的品种。

输出格式
如果可以将这些奶牛分为两个非空的牛群,并使得每个牛群都恰好包含两种奶牛,则输出 Yes,否则输出 No。

数据范围
前 4
个测试点满足 1≤n≤5

所有测试点满足 1≤n≤105

输入样例1:
ababa
输出样例1:
Yes
输入样例2:
aabcc
输出样例2:
Yes
输入样例3:
abbb
输出样例3:
No

思路:这道题就是字符串类型统计的。我们要理解想到满足题目的分法,就必须要至少有2个类型的牛,并且最多4种。
那么我们可以先统计不同类型的牛的有几种,存在count里。如果为大于4或者小于2直接输出No。如果为3,就统计这三种牛的个数,需要至少满足一种牛个数大于等于2,否则不够分。如果为2种,那么2种牛都必须大于等于2个。

#include 

using namespace std;

int main ()
{
    string s;
    cin>>s;
    int n[26] = {0};
    for(auto x : s) n[x - 'a'] ++;
    int count = 0;
    for(int i = 0; i < 26; i ++) if(n[i] != 0) count ++;
    // cout<
    if(count > 4 || count < 2) {
        cout<<"No"<<endl;
        return 0;
    } else if(count == 3){
        int num = 0;
        for(int i = 0; i < 26; i++) {
            if(n[i] < 2 && n[i] != 0) {
                num ++;
            }
            if(num >= 3) {
                cout<<"No"<<endl;
                return 0;
            }
        }
    } else if(count == 2) {
        for(int i = 0; i < 26; i++) {
            if(n[i] < 2 && n[i] != 0) {
                cout<<"No"<<endl;
                return 0;
            }
        }
    }
    cout<<"Yes"<<endl;
    return 0;
}

你可能感兴趣的:(力扣,字符串,c++,算法)