SRM611 Div1 Easy LCMSet

問題概要

2つの数列a,bについて,数列のいくつかの要素に対しLCMを取ることで作れる数の集合は一致するか

解法

aのすべての要素をbから作れ,bのすべての要素を[tex;a]から作れるなら"Equal",そうでないなら"Not equal"
xの約数すべてのLCMを取ってxになるかどうかを考えれば良い.

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(int i=(n);i<(m);i++)
#define REP(i,n) FOR(i,0,n)
#define REPR(i,n) for(int i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<ll,ll>;
using PP=pair<ll,P>;
using vp=vector<P>;
using vpp=vector<PP>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;
ll gcd(ll a,ll b){
    return b==0?a:gcd(b,a%b);
}
ll lcm(ll a,ll b){
    return a/gcd(a,b)*b;
}
class LCMSet{
public:
    string equal(vi a,vi b){
        int n=a.size();
        int m=b.size();
        REP(i,n){
            int l=1;
            REP(j,m){
                if(a[i]%b[j]==0){
                    l=lcm(l,b[j]);
                }
            }
            if(a[i]!=l){
                return "Not equal";
            }
        }
        REP(i,m){
            int l=1;
            REP(j,n){
                if(b[i]%a[j]==0){
                    l=lcm(l,a[j]);
                }
            }
            if(b[i]!=l){
                return "Not equal";
            }
        }
        return "Equal";
    }
};