Submission #779769

#	Time	Username	Problem	Language	Result	Execution time	Memory
779769		vjudge1	Sirni (COCI17_sirni)	C++17	84 / 140	1063 ms	786432 KiB

sirni

#include<bits/stdc++.h>
using namespace std;
#define task "test"
#define pb push_back
#define fi first
#define se second
#define for1(i,j,k) for(int i=j;i<=k;i++)
#define for2(i,j,k) for(int i=j;i>=k;i--)
#define for3(i,j,k,l) for(int i=j;i<=k;i+=l)
#define bit(n,i) ((n>>i)&1)
#define all(x) x.begin(),x.end()
#pragma GCC optimize("Ofast,unroll-loops")
#pragma GCC target("avx,avx2,bmi,bmi2,sse,sse2,sse3,ssse3,sse4,popcnt")
#define int long long
typedef long long ll;
typedef pair<int,int> pii;
typedef double ld;
typedef pair<ld,ld> pdd;
typedef pair<ll,ll> pll;
const ll maxn=1e7+5;
const ll inf=1e18;
const ll mod=1e9+7;
const ll base=311;
int n,nxt[maxn],dd[maxn],par[maxn];
vector<pair<int,pii>> Edge;
int Find(int v) {

    return (par[v] < 0) ?v: (par[v] = Find(par[v]));
}

bool Union(int x, int y) {

    if ((x = Find(x)) == (y = Find(y)))
    {
        return false ;
    }
    if (par[y] < par[x])
    {
        swap(x, y);
    }
    par[x] += par[y];
    par[y] = x;
    return true;
}
void sol()
{
    cin >> n;
    int mx=-1;
    for (int i=1;i<=n;i++)
    {
        int x;cin >> x;

        dd[x]++;
    }
    int ans=0;
    bool ck=0;
    nxt[(int)1e7+1]=-1;
    for2(i,1e7,1)
    {
        par[i]=-1;
        if (dd[i]) nxt[i]=i;
        else nxt[i]=nxt[i+1];
    }
    for (int i=1;i<=1e7;i++)
    {
        //cerr<<"wtf\n";
        if (dd[i])
        {
            if (nxt[i+1]==-1) continue;
            Edge.pb({nxt[i+1]-i,{nxt[i+1],i}});
            for (int j=2*i;j<=1e7;j+=i)
            {
                if (nxt[j]==-1) break;
                Edge.pb({nxt[j]-j,{nxt[j],i}});
            }
        }
    }

    sort(all(Edge));
    for (auto x:Edge)
    {
        //cerr<<x.fi<<' '<<x.se.se<<' '<<x.se.fi<<'\n';
        if (x.se.fi==-1 || x.se.se==-1||!Union(x.se.fi,x.se.se)) continue;
        ans+=x.fi;
    }
    cout << ans;

}
int32_t main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
//    if (fopen(task".inp","r"))
//    {
//        freopen(task".inp","r",stdin);
//        freopen(task".out","w",stdout);
//    }
//    int theta; cin >> theta;
    int t=1;//cin >> t;
    while (t--) sol();
}

Compilation message (stderr)

sirni.cpp: In function 'void sol()':
sirni.cpp:51:9: warning: unused variable 'mx' [-Wunused-variable]
   51 |     int mx=-1;
      |         ^~
sirni.cpp:59:10: warning: unused variable 'ck' [-Wunused-variable]
   59 |     bool ck=0;
      |          ^~

• Subtask 1: 14 / 14
• Subtask 2: 0 / 14
• Subtask 3: 14 / 14
• Subtask 4: 14 / 14
• Subtask 5: 14 / 14
• Subtask 6: 14 / 14
• Subtask 7: 14 / 14
• Subtask 8: 0 / 14
• Subtask 9: 0 / 14
• Subtask 10: 0 / 14