算法题解 - 牛客编程巅峰赛S1第5场 - 黄金&钻石&王者组

A. 完全平方数的尾巴

题目描述

我们把一个能被表示成某个整数的平方的数称为完全平方数。

例如 4 = 2 ∗ 2，16 = 4 ∗ 4，所以4，16是完全平方数。

现在输入一个整数为 x (0 ≤ x ≤ 999)，请聪明的你判断它是不是由某个完全平方数对 1000 取模得到的呢。

示例1

输入

24

输出

true

说明

1024 = 32 ∗ 32
24 = 1024 mod 1000

解法：暴力搜索

思路分析

我们知道完全平方数对 k 取模的值，是 k 个一循环的。即对于 f(x) = (x * x) % k，函数值 k 个一循环。

本题对 1000 取模，因此遍历 1 - 999 的完全平方数模 1000 的值，若都不为 x，则返回 false。

代码实现

public boolean solve (int x) {
   
  for(int i = 1; i < 1000; i++) {
   
    if(i * i % 1000 == x) return true;
  }
  return false;
}

B. 牛牛的字符串

题目描述

牛牛有一个长度为 N 的由小写字母组成的字符串 S，还有一个整数 K。在每一步中，牛牛都可以选择一个位置 i，并在 i 和 i + K 处交换字符（i + K < N）并且 $S_i<S_{i+k}$，即交换之后，新形成的字符串应字典序大于旧字符串。牛牛想尽可能交换尽量多的步数。他想知道最多可以交换多少步呢。

备注:

1 ≤ N ≤ 1e5, 1 ≤ k ≤ N

示例1

输入

"cbexa",2

输出

2

说明

cbexa -> cxeba -> excba 步数为2

解法：分组求解

思路分析

我们看一下交换条件：$S_i<S_{i+k}$。即前一个字符小于后一个字符。

因此对于序列: i, i + k, i + 2k, …。可以交换的次数 = 顺序对的个数。

而字符串中总共有 k 个上述序列。故而求解这 k 个序列的顺序对的个数 之和，即可得出答案。

顺序对个数的求解方法和逆序对个数的求解方法相同，经典的方法有归并排序和离散化树状数组。不过本题是求字符数组的顺序对，因此可以利用 26 大小的哈希数组来求解。

代码分析

public int turn (String s, int k) {
   
  int ans = 0;
  List<Character>[] group = new List[k];
  for (int i = 0; i < k; i++) group[i] = new ArrayList<Character>();
  for (int i = 0; i < s.length(); i++) group[i % k].add(s.charAt(i));
  for (List<Character> list: group) {
   
    int[] arr &#