算法题1（C++）：多线程计算从1到N的所有数字的平方和，其中N为用户输入的正整数（使用多线程来加速）。

编写一个程序，使用多线程计算从1到N的所有数字的平方和，其中N为用户输入的正整数。使用多线程来加速计算过程。

要求: (1)使用 C++的线程库 (std::thread)实现多线程。 (2)程序应该能够处理较大的N值，确保计算结果的准确性和效率。“程序应具有良好的线程同步和数据共享机制，以避免竞争条件和数据不致的问题

#include <iostream>
#include <thread>
#include <vector>
#include <mutex>

std::mutex mtx;  // 创建一个互斥锁
long long sum = 0;  // 创建一个全局变量来存储总和

// 这个函数计算从start到end的平方和
void square_sum(int start, int end) {
    long long partial_sum = 0;  // 创建一个局部变量来存储部分和
    for (int i = start; i <= end; i++) {
        partial_sum += i * i;  // 计算平方和
    }
    std::unique_lock<std::mutex> lock(mtx);  // 使用互斥锁来保护全局变量sum
    sum += partial_sum;  // 将部分和添加到全局变量sum
}

int main() {
    int N;
    std::cout << "Enter a positive integer: ";
    std::cin >> N;

    int num_threads = std::thread::hardware_concurrency();  // 获取硬件支持的并发线程数
    std::vector<std::thread> threads(num_threads);  // 创建一个线程向量

    int chunk_size = N / num_threads;  // 计算每个线程需要处理的数字数量
    for (int i = 0; i < num_threads; i++) {
        int start = i * chunk_size + 1;  // 计算每个线程处理的开始数字
        int end = (i == num_threads - 1) ? N : (i + 1) * chunk_size;  // 计算每个线程处理的结束数字
        threads[i] = std::thread(square_sum, start, end);  // 创建线程并开始计算
    }

    for (auto& t : threads) {
        t.join();  // 等待所有线程完成
    }

    std::cout << "The sum of squares from 1 to " << N << " is " << sum << std::endl;  // 输出结果

    return 0;
}

运行测试：