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 S_i <S_{i+k} Si<Si+k,即交换之后,新形成的字符串应字典序大于旧字符串。牛牛想尽可能交换尽量多的步数。他想知道最多可以交换多少步呢。
备注:
1 ≤ N ≤ 1e5, 1 ≤ k ≤ N
示例1
输入
"cbexa",2
输出
2
说明
cbexa -> cxeba -> excba 步数为2
解法:分组求解
思路分析
我们看一下交换条件: S i < S i + k S_i <S_{i+k} Si<Si+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 =