牛客“迎新春，过大年”多校程序设计竞赛A题

题目描述：

这里有个小trick

当时也看到数据范围的问题了

\(n是1e6 \quad \sum_{i=1}^na[i]<=5e7\)

我们考虑不同的数\(1 \quad 2 \quad ... k-1 \quad k\)

\(sum=(1+k)*k/2\)

不同的数最多有\(n^2\)个

\(O(k^2)\)统计贡献

#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;
}