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;
                    }
}