美团2024年春招第一场笔试【技术】
题目来源:牛客网公司真题_免费模拟题库_企业面试|笔试真题 (nowcoder.com)
1.小美的平衡矩阵
小美拿到了一个n∗n的矩阵,其中每个元素是 0 或者 1。
小美认为一个矩形区域是完美的,当且仅当该区域内 0 的数量恰好等于 1 的数量。
现在,小美希望你回答有多少个i∗i的完美矩形区域。你需要回答1≤i≤n的所有答案。
输入描述:
第一行输入一个正整数,代表矩阵大小。
接下来的行,每行输入一个长度为的 01 串,用来表示矩阵。
输出描述:
输出行,第行输出的完美矩形区域的数量。
示例1
输入例子:
4
1010
0101
1100
0011
输出例子:
0
7
0
1
答案
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
in.nextLine();
int[][] grid = new int[n][n];
int[][] prefixSum = new int[n][n];
for (int i = 0; i < n; i++) {
String str = in.nextLine();
for (int j = 0; j < n; j++) {
grid[i][j] = str.charAt(j) - '0';
prefixSum[i][j] = grid[i][j] + (i > 0 ? prefixSum[i - 1][j] : 0) + (j > 0 ? prefixSum[i][j - 1] : 0) - (i > 0 && j > 0 ? prefixSum[i - 1][j - 1] : 0);
}
}
for (int i = 1; i <= n; i++) {
if (i % 2 == 1) {
System.out.println(0);
} else {
System.out.println(func(prefixSum, n, i));
}
}
in.close();
}
public static int func(int[][] prefixSum, int n, int length) {
int ans = 0;
for (int i = 0; i <= n - length; i++) {
for (int j = 0; j <= n - length; j++) {
int temp = prefixSum[i + length - 1][j + length - 1] - (i > 0 ? prefixSum[i - 1][j + length - 1] : 0) - (j > 0 ? prefixSum[i + length - 1][j - 1] : 0) + (i > 0 && j > 0 ? prefixSum[i - 1][j - 1] : 0);
if (temp == length * length / 2) {
ans++;
}
}
}
return ans;
}
}
2.小美的数组询问
小美拿到了一个由正整数组成的数组,但其中有一些元素是未知的(用 0 来表示)。
现在小美想知道,如果那些未知的元素在区间[l,r]范围内随机取值的话,数组所有元素之和的最小值和最大值分别是多少?
共有q次询问。
输入描述:
第一行输入两个正整数,代表数组大小和询问次数。
第二行输入个整数,其中如果输入的为 0,那么说明是未知的。
接下来的行,每行输入两个正整数 ,代表一次询问。
输出描述:
输出行,每行输出两个正整数,代表所有元素之和的最小值和最大值。
示例1
输入例子:
3 2
1 0 3
1 2
4 4
输出例子:
5 6
8 8
例子说明:
只有第二个元素是未知的。
第一次询问,数组最小的和是 1+1+3=5,最大的和是 1+2+3=6。
第二次询问,显然数组的元素和必然为 8。
答案
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
long n = in.nextLong();
long q = in.nextLong();
in.nextLine();
long sum = 0L;
long zeros = 0L;
for (long i = 0L; i < n; i++) {
long num = in.nextLong();
sum += num;
if (num == 0L) {
zeros++;
}
}
in.nextLine();
for (long i = 0L; i < q; i++) {
long left = in.nextLong();
long right = in.nextLong();
long min = sum + left * zeros;
long max = sum + right * zeros;
System.out.println(min + " " + max);
in.nextLine();
}
in.close();
}
}
3.小美的 MT
MT 是美团的缩写,因此小美很喜欢这两个字母。
现在小美拿到了一个仅由大写字母组成字符串,她可以最多操作k次,每次可以修改任意一个字符。小美想知道,操作结束后最多共有多少个’M’和’T’字符?
输入描述:
第一行输入两个正整数,代表字符串长度和操作次数。
第二行输入一个长度为的、仅由大写字母组成的字符串。
输出描述:
输出操作结束后最多共有多少个'M'和'T'字符。
示例1
输入例子:
5 2
MTUAN
输出例子:
4
例子说明:
修改第三个和第五个字符,形成的字符串为 MTTAM,这样共有 4 个'M'和'T'。
答案
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
long n = in.nextLong();
long k = in.nextLong();
in.nextLine();
String str = in.nextLine();
int ans = 0;
for (int i = 0; i < n; i++) {
if (str.charAt(i) == 'M' || str.charAt(i) == 'T') {
ans++;
} else {
if (k > 0) {
k--;
ans++;
}
}
}
System.out.println(ans);
in.close();
}
}
4.小美的朋友关系
小美认为,在人际交往中,但是随着时间的流逝,朋友的关系也是会慢慢变淡的,最终朋友关系就淡忘了。
现在初始有一些朋友关系,存在一些事件会导致两个人淡忘了他们的朋友关系。小美想知道某一时刻中,某两人是否可以通过朋友介绍互相认识?
事件共有 2 种:
1 u v:代表编号 u 的人和编号 v 的人淡忘了他们的朋友关系。
2 u v:代表小美查询编号 u 的人和编号 v 的人是否能通过朋友介绍互相认识。
注:介绍可以有多层,比如 2 号把 1 号介绍给 3 号,然后 3 号再把 1 号介绍给 4 号,这样 1 号和 4 号就认识了。
输入描述:
第一行输入三个正整数,代表总人数,初始的朋友关系数量,发生的事件数量。
接下来的行,每行输入两个正整数,代表初始编号的人和编号的人是朋友关系。
接下来的行,每行输入三个正整数,含义如题目描述所述。
保证至少存在一次查询操作。
输出描述:
对于每次 2 号操作,输出一行字符串代表查询的答案。如果编号 u 的人和编号 v 的人能通过朋友介绍互相认识,则输出"Yes"。否则输出"No"。
示例1
输入例子:
5 3 5
1 2
2 3
4 5
1 1 5
2 1 3
2 1 4
1 1 2
2 1 3
输出例子:
Yes
No
No
例子说明:
第一次事件,1 号和 5 号本来就不是朋友,所以无事发生。
第二次事件是询问,1 号和 3 号可以通过 2 号的介绍认识。
第三次事件是询问,显然 1 号和 4 号无法互相认识。
第四次事件,1 号和 2 号淡忘了。
第五次事件,此时 1 号无法再经过 2 号和 3 号互相认识了。
答案1(不能通过全部用例)
import java.io.*;
import java.util.*;
public class Main {
public static Map<Integer, Map<Integer, Boolean>> map;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String[] str = br.readLine().split(" ");
int n = Integer.parseInt(str[0]);
int m = Integer.parseInt(str[1]);
int q = Integer.parseInt(str[2]);
map = new HashMap<>(n);
for (int i = 0; i < m; i++) {
str = br.readLine().split(" ");
int a = Integer.parseInt(str[0]);
int b = Integer.parseInt(str[1]);
Map<Integer, Boolean> mapA = map.getOrDefault(a, new HashMap<>());
Map<Integer, Boolean> mapB = map.getOrDefault(b, new HashMap<>());
mapA.put(b, true);
mapB.put(a, true);
map.put(a, mapA);
map.put(b, mapB);
}
for (int i = 0; i < q; i++) {
str = br.readLine().split(" ");
int op = Integer.parseInt(str[0]);
int a = Integer.parseInt(str[1]);
int b = Integer.parseInt(str[2]);
if (op == 1) {
Map<Integer, Boolean> mapA = map.getOrDefault(a, new HashMap<>());
Map<Integer, Boolean> mapB = map.getOrDefault(b, new HashMap<>());
mapA.put(b, false);
mapB.put(a, false);
map.put(a, mapA);
map.put(b, mapB);
} else {
if (dfs(a, b, new boolean[n + 1])) {
System.out.println("Yes");
} else {
System.out.println("No");
}
}
}
br.close();
}
public static boolean dfs(int parent, int target, boolean[] visited) {
if (visited[parent]) {
return false;
}
Map<Integer, Boolean> mapParent = map.get(parent);
if (mapParent == null) {
return false;
}
Boolean b = mapParent.get(target);
if (b != null) {
return b;
}
visited[parent] = true;
for (Map.Entry<Integer, Boolean> entry : mapParent.entrySet()) {
if (entry.getKey() != parent && entry.getValue() && dfs(entry.getKey(), target, visited)) {
return true;
}
}
visited[parent] = false;
return false;
}
}
答案2(可以通过全部用例)
【算法】美团2024年春招第一场笔试【技术】- Q4小美的朋友关系【反向并查集+离散化】_java反向并查集解释-CSDN博客
5.小美的区间删除
小美拿到了一个大小为n的数组,她希望删除一个区间后,使得剩余所有元素的乘积末尾至少有k个 0。小美想知道,一共有多少种不同的删除方案?
输入描述:
第一行输入两个正整数。
第二行输入个正整数,代表小美拿到的数组。
输出描述:
一个整数,代表删除的方案数。
示例1
输入例子:
5 2
2 5 3 4 20
输出例子:
4
例子说明:
第一个方案,删除[3]。
第二个方案,删除[4]。
第三个方案,删除[3,4]。
第四个方案,删除[2]。
答案1(不能通过全部用例)
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int k = in.nextInt();
in.nextLine();
long num;
long[] product = new long[n];
for (int i = 0; i < n; i++) {
num = in.nextLong();
if (i == 0) {
product[i] = num;
} else {
product[i] = product[i - 1] * num;
}
}
long ans = 0L;
for (int i = 0; i < n; i++) {
for (int j = i; j < n; j++) {
if (check(product, n, k, i, j)) {
ans++;
}
}
}
System.out.println(ans);
in.close();
}
public static boolean check(long[] product, int n, int k, int left, int right) {
if (product[right] == 0L) {
return k == 1;
}
long temp = product[n - 1] / product[right];
if (left > 0) {
temp *= product[left - 1];
}
for (int i = 0; i < k; i++) {
long a = temp % 10L;
if (a != 0L) {
return false;
}
temp /= 10L;
}
return true;
}
}
答案2(可以通过全部用例)
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int k = in.nextInt();
in.nextLine();
long num;
long[] pre2 = new long[n + 1];
long[] pre5 = new long[n + 1];
for (int i = 0; i < n; i++) {
num = in.nextLong();
long cntFactor2 = 0L, cntFactor5 = 0L;
for (long j = num; j % 2 == 0; j /= 2) {
cntFactor2++;
}
for (long j = num; j % 5 == 0; j /= 5) {
cntFactor5++;
}
pre2[i + 1] += (pre2[i] + cntFactor2);
pre5[i + 1] += (pre5[i] + cntFactor5);
}
long ans = 0L;
for (int i = 0, j = 0; i < n; i++) {
while (j < n) {
long remain2 = pre2[n] - pre2[j + 1] + pre2[i];
long remain5 = pre5[n] - pre5[j + 1] + pre5[i];
if (Math.min(remain2, remain5) < k) {
break;
}
j++;
}
ans += Math.max(j - i, 0L);
}
System.out.println(ans);
in.close();
}
}