(新卷,100分)- 检查是否存在满足条件的数字组合（Java & JS & Python & C）

(新卷,100分)- 检查是否存在满足条件的数字组合（Java & JS & Python & C）

题目描述

给定一个正整数数组，检查数组中是否存在满足规则的数字组合

规则：A = B + 2C

输入描述

第一行输出数组的元素个数。

接下来一行输出所有数组元素，用空格隔开。

输出描述

如果存在满足要求的数，在同一行里依次输出规则里A/B/C的取值，用空格隔开。

如果不存在，输出0。

备注

用例

题目解析

由于数组长度较小（3~100），即使采用三重循环的暴力解法也不会超时。

解题步骤如下：

备注说明：

• 题目保证每组数字最多只有一组符合条件的解
• 因此一旦找到符合条件的解，即可立即返回结果

Java算法源码

import java.util.Arrays;
import java.util.Scanner;

public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);

int n = Integer.parseInt(sc.nextLine());
Integer[] arr =
Arrays.stream(sc.nextLine().split(" ")).map(Integer::parseInt).toArray(Integer[]::new);

System.out.println(getResult(n, arr));
}

public static String getResult(int n, Integer[] arr) {
Arrays.sort(arr, (a, b) -> b – a);

for (int i = 0; i < n; i++) {
for (int j = i + 1; j < n; j++) {
for (int k = j + 1; k < n; k++) {
if (arr[i] == arr[j] + 2 * arr[k]) {
return arr[i] + " " + arr[j] + " " + arr[k];
}
if (arr[i] == arr[k] + 2 * arr[j]) {
return arr[i] + " " + arr[k] + " " + arr[j];
}
}
}
}

return "0";
}
}

JS算法源码

/* JavaScript Node ACM模式 控制台输入获取 */
const readline = require("readline");

const rl = readline.createInterface({
input: process.stdin,
output: process.stdout,
});

const lines = [];
rl.on("line", (line) => {
lines.push(line);

if (lines.length === 2) {
let n = parseInt(lines[0]);
let arr = lines[1].split(" ").map(Number);

console.log(getResult(n, arr));

lines.length = 0;
}
});

function getResult(n, arr) {
arr.sort((a, b) => b – a);

for (let i = 0; i < n; i++) {
for (let j = i + 1; j < n; j++) {
for (let k = j + 1; k < n; k++) {
if (arr[i] == arr[j] + 2 * arr[k]) {
return `${arr[i]} ${arr[j]} ${arr[k]}`;
}

if (arr[i] == arr[k] + 2 * arr[j]) {
return `${arr[i]} ${arr[k]} ${arr[j]}`;
}
}
}
}

return "0";
}

Python算法源码

# 输入获取
n = int(input())
arr = list(map(int, input().split()))

# 算法入口
def getResult():
arr.sort(reverse=True)

for i in range(n):
for j in range(i + 1, n):
for k in range(j + 1, n):
if arr[i] == arr[j] + 2 * arr[k]:
return f"{arr[i]} {arr[j]} {arr[k]}"

if arr[i] == arr[k] + 2 * arr[j]:
return f"{arr[i]} {arr[k]} {arr[j]}"

return "0"

# 算法调用
print(getResult())

C算法源码

#include <stdio.h>
#include <stdlib.h>

int cmp(const void* a, const void* b) {
return (*(int*) b) – (*(int*) a);
}

int main() {
int n;
scanf("%d", &n);

int nums[n];
for(int i=0; i<n; i++) {
scanf("%d", &nums[i]);
}

qsort(nums, n, sizeof(int), cmp);

for(int i=0; i<n; i++) {
for(int j=i+1; j<n; j++) {
for(int k=j+1; k<n; k++) {
if(nums[i] == nums[j] + 2 * nums[k]) {
printf("%d %d %d\\n", nums[i], nums[j], nums[k]);
return 0;
}

if(nums[i] == nums[k] + 2 * nums[j]) {
printf("%d %d %d\\n", nums[i], nums[k], nums[j]);
return 0;
}
}
}
}

puts("0");

return 0;
}