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

题目描述:
image

这里有个小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;
}

posted @ 2024-02-14 00:50  cxy8  阅读(23)  评论(0编辑  收藏  举报