题目描述:
这里有个小trick
当时也看到数据范围的问题了
n
是
1
e
6
∑
i
=
1
n
a
[
i
]
<
=
5
e
7
n是1e6 \quad \sum_{i=1}^na[i]<=5e7
n是1e6∑i=1na[i]<=5e7
我们考虑不同的数
1
2
.
.
.
k
−
1
k
1 \quad 2 \quad ... k-1 \quad k
12...k−1k
s
u
m
=
(
1
+
k
)
∗
k
/
2
sum=(1+k)*k/2
sum=(1+k)∗k/2
不同的数最多有
n
2
n^2
n2个
O
(
k
2
)
O(k^2)
O(k2)统计贡献
#include <bits/stdc++.h>
#define int long long
#define rep(i,a,b) for(int i = (a); i <= (b); ++i)
#define fep(i,a,b) for(int i = (a); i >= (b); --i)
#define pii pair<int, int>
#define ll long long
#define db double
#define endl '\n'
#define x first
#define y second
#define pb push_back
using namespace std;
void solve()
{
int n;cin>>n;
map<int,int>p;
//map去重
rep(i,1,n){
int x;cin>>x;
p[x]++;
}
//统计贡献
int ans=0;
const int mod=998244353;
for(auto [i,cnt1]:p){
for(auto [j,cnt2]:p){
ans=(ans+__gcd(i,j)*(__builtin_popcount(i)+__builtin_popcount(j))%mod*cnt1%mod*cnt2%mod)%mod;
}
}
cout<<ans<<endl;
}
signed main()
{
ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
// freopen("1.in", "r", stdin);
int _;
// cin>>_;
// while(_--)
solve();
return 0;
}