1534

include/solution/1534.h

@@ -0,0 +1,15 @@
+//
+// Created by xfj12 on 2025/4/14.
+//
+
+#ifndef INC_1534_H
+#define INC_1534_H
+#ifdef __cplusplus
+extern "C"
+{
+#endif
+    int countGoodTriplets(int *arr, int arrSize, int a, int b, int c);
+#ifdef __cplusplus
+}
+#endif
+#endif // INC_1534_H

src/1534.c

@@ -0,0 +1,17 @@
+//
+// Created by xfj12 on 2025/4/14.
+//
+#include <math.h>
+#include <solution/1534.h>
+int countGoodTriplets(int *arr, int arrSize, int a, int b, int c)
+{
+    int count = 0;
+
+    for (int i = 0; i < arrSize; i++)
+        for (int j = i + 1; j < arrSize; j++)
+            for (int k = j + 1; k < arrSize; k++)
+                if (abs(arr[i] - arr[j]) <= a && abs(arr[j] - arr[k]) <= b && abs(arr[i] - arr[k]) <= c)
+                    count++;
+
+    return count;
+}

tests/test_1534.cpp

@@ -0,0 +1,23 @@
+#include <gtest/gtest.h>
+#include <solution/1534.h>
+
+TEST(CountGoodTripletsTest, Test1)
+{
+    int arr[] = {3, 0, 1, 1, 9, 7};
+    int arrSize = sizeof(arr) / sizeof(arr[0]);
+    int a = 7, b = 2, c = 3;
+    int expected = 4;
+
+    EXPECT_EQ(countGoodTriplets(arr, arrSize, a, b, c), expected);
+}
+
+// 测试用例 2
+TEST(CountGoodTripletsTest, Test2)
+{
+    int arr[] = {1, 1, 2, 2, 3};
+    int arrSize = sizeof(arr) / sizeof(arr[0]);
+    int a = 0, b = 0, c = 1;
+    int expected = 0;
+
+    EXPECT_EQ(countGoodTriplets(arr, arrSize, a, b, c), expected);
+}