面经: 腾讯机器学习算法岗

求一次函数与二次函数围成的面积

程序

import sys
import numpy as np
from scipy import integrate  # 计算积分

def calcu_s(A, B, C):
    # c1是b^2, c2是4ac
    c1 = 4 * (A ** 2)
    c2 = 8 * A * B * C
    # 判断 b^2 - 4ac是否大于零（确保有两个交点，才能计算面积）
    if (c1 - c2) > 0:
        # 联立方程组 y^2 = A*x^2 与 y = B * x + C，求出两个交点坐标
        point1_x = (1 / (2 * (B ** 2) + (10 ** -7))) * (2 * (A - B*C) + np.sqrt(c1 - c2))  # 注意0除问题， 分母添加(10 ** -7)
        point1_y = B * point1_x + C
        point2_x = (1 / (2 * (B ** 2) + (10 ** -7))) * (2 * (A - B*C) - np.sqrt(c1 - c2))
        point2_y = B * point2_x + C
        # 计算直线与y轴围成的梯形面积，取绝对值
        s1 = (abs(point1_x) + abs(point2_x)) * abs(point1_y - point2_y) * 0.5
        print(s1)
        # 计算二次函数与y轴围城的面积， 取绝对值
        def f(x):
            return (1 / (2 * A + (10 ** -7))) * (x ** 2)
        s2, err = integrate.quad(f, point1_y, point2_y)
        s2 = abs(s2)
        print(s2)
        # 计算两个函数围成的面积
        return s1 - s2
    else:
        return 0

if __name__ == "__main__":
    # 读取第一行的n
    n = int(sys.stdin.readline().strip())
    ans = []
    for i in range(n):
        ret = []
        # 读取每一行
        line = sys.stdin.readline().strip()
        # 把每一行的数字分隔后转化成int列表
        values = list(map(int, line.split()))
        for v in values:
            ret.append(v)
        ans.append(ret)  # 将每一行转化为嵌套列表的格式， 每一个内嵌列表对应一组参数
    print(ans)
    for i in range(len(ans)):
        # 遍历每一组参数
        final = calcu_s(ans[i][0], ans[i][1], ans[i][2])
        print(final)

收获总结

(1) scipy.integrate 函数计算积分(曲线面积)

(2) 输入数据获取