牛客编程巅峰赛S2第12场

牛牛和好数

题目描述

若一个数的首位和末位相等，则定义这个数为“好数”。
例如：1231、4512394是好数，而12345、768740则不是好数。
请你编写一个函数，判断是不是好数。如果是好数则返回true，否则返回false。

示例1

输入

1231

返回值

true

说明 >首位和末位都是1，相等。

示例2

输入

4

返回值

true

说明

首位和末位都是4，相等。

示例3

输入

100

返回值

false

说明

首位是1，末位是0，不相等。

备注:

1≤x≤10^9

签到题：
代码如下：

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 判断x是不是好数
     * @param x int整型 待判断的数
     * @return bool布尔型
     */
    bool judge(int x) {
        // write code here
        return x%10 == to_string(x)[0]-'0';
        
    }
};

统计好元组

题目描述

现在给定一个数组arr，和a,b两个数字，你要做的就是找到（i，j，k）。且满足

1. 0 <= i < j < k < arr.size()
2. |arr[i] - arr[j]| <= a
3. |arr[j] - arr[k]| <= b
   统计满足条件的个数并返回(最后结果可能很大，请取1000000007的余数)。

示例1

输入

[7,1,8,9,0],3,3

返回值

1

说明

只有（7，8，9）符合要求

备注:

arr.size() <= 5000
其余变量均<=1e9

代码如下：

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 
     * @param arr int整型vector 
     * @param a int整型 
     * @param b int整型 
     * @return int整型
     */
    int countTriplets(vector<int>& arr, int a, int b) {
        // write code here
        int mod = 1000000007;
        int ans = 0;
        int n = arr.size();
        for(int j=0;j<n;j++){
            int x = 0, y = 0;
            for(int i = 0; i < j; i++){
                if(abs(arr[i] - arr[j]) <= a){
                    x++;
                }
            }
            for(int k = j + 1; k < n; k++){
                if(abs(arr[k] - arr[j]) <= b){
                    y++;
                }
            }
            ans = (ans + x * y) % mod; 
        }
        return ans;
    }
};