1. 程式人生 > >CF851 D 枚舉 思維

CF851 D 枚舉 思維

https 多少 tps 使用 set fine %d namespace ems

給出n個數,你可以對每個數把它變為0,或者增加1,分別需要花費x, y。問把所有數的GCD變為不為1的最小花費是多少。

n的範圍5x1e5,a[i]的範圍1e6。

開始想通過枚舉最終gcd值,然後通過判左右個數以及消費來二分,顯然是愚蠢的想法,因為一個數在不同模數下余數並不單調阿!

實際上是枚舉gcd值,首先a[i]只有1e6範圍,預處理前綴和:cnt[i]表示前a[] < i的個數和,sum[i] 比i小的所有a[]的和。

這樣在枚舉gcd的倍數值時,只要找到gcd範圍內的一個劃分,小於該劃分的數的余數使用消去消費<增加該數到gcd的倍數的消費,那麽只要計算gcd的所有倍數,就能得到該gcd作為最終因子的花費了。

/** @Date    : 2017-09-06 19:32:17
  * @FileName: D.cpp
  * @Platform: Windows
  * @Author  : Lweleth ([email protected])
  * @Link    : https://github.com/
  * @Version : $Id$
  */
#include <bits/stdc++.h>
#define LL long long
#define PII pair<int ,int>
#define MP(x, y) make_pair((x),(y))
#define fi first
#define se second
#define PB(x) push_back((x))
#define MMG(x) memset((x), -1,sizeof(x))
#define MMF(x) memset((x),0,sizeof(x))
#define MMI(x) memset((x), INF, sizeof(x))
using namespace std;

const int INF = 0x3f3f3f3f;
const int N = 1e6+20;
const double eps = 1e-8;

LL n, x, y;
LL sum[N*2], cnt[N*2];
int main()
{
	while(cin >> n >> x >> y)
	{
		MMF(cnt);
		MMF(sum);
		for(int i = 0; i < n; i++)
		{
			LL t;
			scanf("%lld", &t);
			cnt[t]++;
			sum[t] += t;
			/*if(n <= 1)
			{
				printf("%d\n", t==1?min(x,y):0);
				return 0;
			}*/
		}

		for(int i = 1; i < N*2; i++)
			cnt[i] += cnt[i - 1], sum[i] += sum[i - 1];

		LL ans = 1e16;
		for(LL i = 2; i <= 1000000; i++)
		{
			LL t = 0;
			for(LL j = i; j < 1000000 + i; j+=i)
			{
				LL ma = max(j - i + 1, j - x / y);
				t += (cnt[ma - 1] - cnt[j - i]) * x;//直接消去
				t += ((cnt[j] - cnt[ma - 1]) * j - (sum[j] - sum[ma - 1])) * y;
			}
			if(t < ans && t >= 0)
				ans = t;
			
		}
		printf("%lld\n", ans);
	}
    return 0;
}

CF851 D 枚舉 思維