【两次过】【2018网易】彩色的砖块

最新推荐文章于 2021-07-16 20:48:32 发布

小马哥MAX

最新推荐文章于 2021-07-16 20:48:32 发布

阅读量191

点赞数

CC 4.0 BY-SA版权

分类专栏：牛客真题

本文链接：https://blog.youkuaiyun.com/majichen95/article/details/90199669

牛客真题专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种关于彩色砖块排列的算法问题，通过统计不同颜色砖块的数量，快速判断并计算出满足条件的排列方式总数。对于仅包含一到两种颜色的砖块，算法能给出准确的排列数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

小易有一些彩色的砖块。每种颜色由一个大写字母表示。各个颜色砖块看起来都完全一样。现在有一个给定的字符串s,s中每个字符代表小易的某个砖块的颜色。小易想把他所有的砖块排成一行。如果最多存在一对不同颜色的相邻砖块,那么这行砖块就很漂亮的。请你帮助小易计算有多少种方式将他所有砖块排成漂亮的一行。(如果两种方式所对应的砖块颜色序列是相同的,那么认为这两种方式是一样的。)
例如: s = "ABAB",那么小易有六种排列的结果:
"AABB","ABAB","ABBA","BAAB","BABA","BBAA"
其中只有"AABB"和"BBAA"满足最多只有一对不同颜色的相邻砖块。

输入描述:

输入包括一个字符串s,字符串s的长度length(1 ≤ length ≤ 50),s中的每一个字符都为一个大写字母(A到Z)。

输出描述:

输出一个整数,表示小易可以有多少种方式。

示例1

输入

ABAB

输出

解题思路：

一开始想的很复杂，利用dfs类似全排列的方式求解，时间复杂度高，只通过了50%.

后来看答案才发现，这是有规律的

对每种颜色砖块个数做统计：

如果出现了两种以上的颜色，肯定最少有两种相邻，直接输出0.
如果出现了一种，那就只有一种。
如果出现了两种，那就只有两种。

import java.util.*;

public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        char[] str = sc.nextLine().toCharArray();
        Set<Character> set = new HashSet<>();
        int res = 0;
        
        for(char c : str)
            set.add(c);
        
        if(set.size() == 1)
            res = 1;
        else if(set.size() == 2)
            res = 2;
        else
            res = 0;
        
        System.out.println(res);
    }
}