数论 B - Products of Min-Max

原创于 2021-04-06 18:17:41 发布 · 171 阅读

0 ·

CC 4.0 BY-SA版权

计算几何专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一段C++代码，用于计算给定数组中元素的平方和，并对结果取模。通过排序数组和使用滚动数组技巧简化计算过程。适合学习者理解数组操作和优化算法。

添加链接描述

思路

添加链接描述

code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 3e5;
const ll mod = 998244353;
ll n, m, a[N];
int main () 
{
    cin >> n;
    for(ll i = 1; i <= n; i++)
        cin >> a[i];
    sort(a+1, a+n+1);
    ll ans = 0, temp = 0;
    for(ll i = 1; i <= n; i++)
    {
        ans = (ans + a[i]*a[i]) % mod;
        ans = (ans + a[i] * temp) % mod;
        temp = (2 * temp + a[i]) % mod;
    }
    cout << ans%mod << endl;
    return 0;
}