Cod sursa(job #2268424)

Utilizator DavidLDavid Lauran DavidL Data 24 octombrie 2018 19:57:01
Problema Multiplu Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.67 kb
#include <bits/stdc++.h>
#define pb push_back
#define ll long long
using namespace std;
ifstream fi("multiplu.in");
ofstream fo("multiplu.out");

ll a, b, lcm;
ll curr;
vector <ll> cif, ult, rez;
unordered_map <ll, bool> avem;
queue < pair <ll, ll> > Q; /// (restul, momentul)

ll gcd(ll a, ll b)
{
    if (b == 0)
        return a;
    return gcd(b, a % b);
}

signed main()
{
    fi >> a >> b;

    if (a + b == 1)
    {
        fo << 1;
        return 0;
    }

    lcm = a * b / gcd(a, b);

    Q.push({1, 0});
    avem[1] = 0;
    ult.pb(-1);
    cif.pb(1);

    ll deacolo = -1;

    while (!Q.empty())
    {
        pair <ll, ll> nod = Q.front();
        Q.pop();

        ll r = nod.first, t = nod.second;

        /// bagam 0 la final
        if (!avem[r * 10 % lcm])
        {
            curr++;
            Q.push({r * 10 % lcm, curr});
            avem[r * 10 % lcm] = 1;
            ult.pb(t);
            cif.pb(0);

            if (r * 10 % lcm == 0)
            {
               deacolo = curr;
               break;
            }
        }

        /// bagam 1
        if (!avem[(r * 10 + 1) % lcm])
        {
            curr++;
            Q.push({(r * 10 + 1) % lcm, curr});
            avem[(r * 10 + 1) % lcm] = 1;
            ult.pb(t);
            cif.pb(1);

            if ((r * 10 + 1) % lcm == 0)
            {
               deacolo = curr;
               break;
            }
        }
    }

    while (deacolo != -1)
    {
        rez.pb(cif[deacolo]);
        deacolo = ult[deacolo];
    }

    reverse(rez.begin(), rez.end());
    for (auto x: rez)
        fo << x;

    return 0;
}