CF1614D1 - Divan and Kostomuksha (Easy version) 题解

链接：https://codeforces.com/contest/1614/problem/D1
难度：*2100
标签：dp number theory

题意

给你 $n$ 个数，求出 $\sum _{i=1}^n\gcd (a_1,a_2,a_3,...,a_i)$

题解

定义 $cnt[i]$：数组 $a$ 中是 $i$ 的倍数的元素个数

定义 $dp[i]$：表示目前 $\gcd =i$，前缀 $\gcd$ 和的最大值。

转移：
$$dp[i]={\max }_{j\mathrm{mod}i=0,i\ne j}(dp[i],dp[j]+i\ast (cnt[j]-cnt[i]))$$

C++ 代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=5e6+5;
int n;
int v[maxn];
int cnt[maxn];
int dp[maxn];
signed main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>v[i];
		cnt[v[i]]++; 
	}
	for(int i=1;i<maxn;i++){
		for(int j=2*i;j<maxn;j+=i){
			cnt[i]+=cnt[j];
		}
	}
	int mx=0;
	for(int i=maxn-1;i>=1;i--){
		dp[i]=i*cnt[i];
		for(int j=2*i;j<maxn;j+=i){
			dp[i]=max(dp[i],dp[j]+i*(cnt[i]-cnt[j]));
		}
		mx=max(dp[i],mx);
	}
	cout<<mx<<endl;
	return 0;
}