做题记录整理贪心4 P6187 [NOI Online #1 提高组] 最小环（2022/9/13）

P6187 [NOI Online #1 提高组] 最小环

就是要想到让大的乘大的，小的乘小的，最后记忆化就完事

#include<bits/stdc++.h>
using namespace std;
#define for1(i,a,b) for(ll i = a;i <=b;i++)
#define ll long long
using namespace std;
ll a[500005],ji[500005];
int gcd(int a,int b){
	return b==0?a:gcd(b,a%b);
}
int n,m,t,p;
int main()
{
	cin>>n>>m;
	for1(i,1,n) cin>>a[i];
	sort(a+1,a+n+1);
	for1(i,1,m)
	{
		ll ans=0;
		int k;
		cin>>k;
		if(k==0)
		{
			for1(i,1,n) ans+=a[i]*a[i];
			printf("%lld\n",ans);
			continue;
		}
		t=gcd(n,k),p=n/t;
		if(ji[p])
		{
			printf("%lld\n",ji[p]);
			continue;
		}
		for(int i=1;i<=n;i+=p){
			for1(j,0,p-3) ans+=a[i+j]*a[i+j+2];
			ans+=(a[i]*a[i+1]+a[i+p-1]*a[i+p-2]);
		}
		printf("%lld\n",ans);
		ji[p]=ans;
	}
	return 0;
}