【春招笔试】2024-05-11-蚂蚁金服-三语言题解

🍭 大家好这里是清隆学长 ，一枚热爱算法的程序员

✨ 本系列打算持续跟新蚂蚁近期的春秋招笔试题汇总～

👏 感谢大家的订阅➕ 和 喜欢💗

✨ 笔试合集传送们 -> 🧷春秋招笔试合集

⌚️ 01.最大化权值

问题描述

LYA 定义一个数组的权值为正数元素的数量减去负数元素的数量。例如， 的权值为 ，而 的权值为 。

现在 LYA 拿到了一个数组，她准备选择恰好 个元素进行取反（同一个元素最多只能取反一次），目标是使得数组的权值尽可能大。你能帮她计算出操作后数组权值的最大值吗？

输入格式

第一行包含两个正整数 和 ，分别表示数组的长度和选择的元素数量。

第二行包含 个整数 ，表示给定的数组。

输出格式

输出一个整数，表示操作后数组权值的最大值。

样例输入

5 3
-5 1 4 -2 3

样例输出

3

数据范围

• 
• 

可以统计数组中负数、正数和零的个数，分别记为 、 和 。

如果 ，即负数的个数大于等于可以取反的元素数量，那么我们只需要将 个负数取反，最终的权值即为 。

如果 ，那么我们需要先将所有负数取反，此时还剩下 次取反操作。如果 ，即剩下的取反次数小于等于零的个数，那么我们无需进行更多操作，最终的权值为 。否则，我们需要将 个正数取反。

参考代码

• Python

n, k = map(int, input().split())

l, r, c = 0, 0, 0
for x in (list(map(int, input().split()))):
    if x < 0:
        l += 1
    elif x > 0:
        r += 1
    else:
        c += 1

t = l
l -= k
if l >= 0:
    r += k
    print(r - l)
else:
    l = 0
    k -= t
    k -= c
    r += t
    if k <= 0:
        print(r)
    else:
        r -= k
        l += k
        print(r - l)

• Java

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int k = scanner.nextInt();

        int l = 0, r = 0, c = 0;
        for (int i = 0; i < n; i++) {
            int x = scanner.nextInt();
            if (x < 0)
                l++;
            else if (x > 0)
                r++;
            else
                c++;
        }

        int t = l;
        l -= k;
        if (l >= 0) {
        	r += k;
            System.out.println(r - l);
        } else {
            l = 0;
            k -= t;
            k -= c;
            r += t;
            if (k <= 0)
                System.out.println(r);
            else {
                r -= k;
                l += k;
                System.out.println(r - l);
            }
        }
    }
}

• Cpp

#include <iostream>
using namespace std;

int main() {
    int n, k;
    cin >> n >> k;

    int l = 0, r = 0, c = 0;
    for (int i = 0; i < n; i++) {
        int x;
        cin >> x;
        if (x < 0)
            l++;
        else if (x > 0)
            r++;
        else
            c++;
    }

    int t = l;
    l -= k;
    if (l >= 0) {
        r += k;
        cout << r - l << endl;
    } else {
        l = 0;
        k -= t;
        k -= c;
        r += t;
        if (k <= 0)
            cout << r << endl;
        else {
            r -= k;
            l += k;
            cout << r - l << endl;
        }
    }

    return 0;
}

📱 02.等腰梯形判定

问题描述

LYA 在平面上找到了四个点的坐标，想要判断这四个点能否构成一个等腰梯形。等腰梯形的定义为：四条边分为两对，有一对边平行且不相等，另一对边相等且不平行。

给定四个点的坐标，请帮助 LYA 判断它们是否能构成一个等腰梯形。

输入格式

第一行包含一个正整数 ，表示询问的次数。

接下来的 行，每行包含 个整数 ，分别表示四个点的坐标。

输出格式

对于每个询问，如果给定的四个点能构成等腰梯形，则输出 "Yes"，否则输出 "No"。

样例输入

3
0 0 2 4 5 0 3 4
0 0 1 4 2 0 3 4
0 1 2 1 3 2 3 4

样例输出

Yes
No
Yes

数据范围

• 
• 

题解

判断四个点是否能构成等腰梯形，可以按照以下步骤进行：

1. 计算四条边的长度，分别记为 ，以及两条对角线的长度 。
2. 判断是否存在一对平行且不相等的边，以及另一对相等且不平行的边：
   • 如果 且 ，则判断 或者 是否成立。
   • 如果 且 ，则判断 或者 是否成立。
   • 如果 且 ，则判断 或者 是否成立。
3. 如果满足上述条件之一，则给定的四个点能构成等腰梯形，否则不能。

其中，判断两条线段是否平行可以通过比较它们的斜率是否相等来实现。

参考代码

• Cpp

#include <iostream>
#include <cmath>
using namespace std;

struct Point {
    int x, y;
};

double dist(Point a, Point b) {
    return sqrt(pow(a.x - b.x, 2) + pow(a.y - b.y, 2));
}

bool isParallel(Point a, Point b, Point c, Point d) {
    return (b.y - a.y) * (d.x - c.x) == (d.y - c.y) * (b.x - a.x);
}

bool isIsosceles(Point a, Point b, Point c,