JZOJ 5445. 【NOIP2017提高A组冲刺11.2】失格

Description

胆小鬼连幸福都会害怕，碰到棉花都会受伤，有时还被幸福所伤。
——太宰治《人间失格》

回顾我的一生，一共有n个事件，每一个事件有一个幸福值p_i。
我想用n-1条线把所有的事件连起来，变成一个连通块。一条连接了事件x和事件y的线会产生min(p_x mod p_y,p_y mod p_x)的喜悦值。
日日重复同样的事，遵循着与昨日相同的惯例，若能避开猛烈的狂喜，自然也不会有悲痛的来袭。因此，我想知道连接起来之后产生喜悦值最小是多少。

Input

文件第一行有一个正整数n。
接下来n行，每行一个正整数p_i。

Output

输出只有一行，表示最小的喜悦值。

Sample Input

输入1：

4
2
6
3
11

输入2：

4
1
2
3
4

输出3：

3
4
9
15

Sample Output

输出1：

1

输出2：

0

输出3：

4

Data Constraint

对于30%的数据，保证1<=n<=10^3。
对于另外40%的数据，保证1<=p_i<=10^6。
对于100%的数据，保证1<=n<=10^5,1<=p_i<=10^7。

Solution

• 枚举一下余数，贪心地跑一跑即可。

Code

#include<cstdio>#include<algorithm>using namespace std;const int N=1e7+1;int mx;long long ans;int a[100001],f[N],size[N];bool bz[N];inline int read(){    int X=0,w=1; char ch=0;    while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();    return X*w;}inline int max(int x,int y){    return x>y?x:y;}inline int get(int x){    return f[x]==x?x:f[x]=get(f[x]);}inline bool merge(int x,int y){    if(get(x)==get(y)) return false;    if(size[f[x]]>size[f[y]]) swap(x,y);    size[f[y]]+=size[f[x]];    f[f[x]]=f[y];    return true;}int main(){    int n=read(),m=0;    for(int i=1;i<=n;i++)    {        int x=read();        mx=max(mx,x);        if(!f[x]) a[++m]=x;        size[f[x]=x]=1;    }    for(int k=0,num=1;k<=mx;k++)        for(int i=1;i<=m;i++)            if(!bz[a[i]])                for(int j=1;a[i]*j+k<=mx;j++)                    if(f[a[i]*j+k] && merge(a[i],a[i]*j+k))                    {                        ans+=k;                        if(++num==m) return 0&printf("%lld",ans);                        if(!k) bz[a[i]*j]=true;                    }}