제출 #837922

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
837922		tfgs	Zamjena (COCI18_zamjena)	C++17	14 / 70	55 ms	65536 KiB

zamjena

#include <bits/stdc++.h>
using namespace std;
using pll = pair<long long, long long>;
#define ll long long
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define pb push_back
#define eb emplace_back
#define vi vector<ll>
#define vb vector<bool>
#define f0(i,a) for(ll i=0;i<(a);i++)
#define f1(i,a) for(ll i=1;i<(a);i++)
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
#define f first
#define s second
template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << '(' << p.first << ", " << p.second << ')'; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> ostream& operator<<(ostream &os, const T_container &v) { os << '{'; string sep; for (const T &x : v) os << sep << x, sep = ", "; return os << '}'; }
// http://www.open-std.org/jtc1/sc22/wg21/docs/papers/2016/p0200r0.html
template<class Fun> class y_combinator_result {
    Fun fun_;
public:
    template<class T> explicit y_combinator_result(T &&fun): fun_(std::forward<T>(fun)) {}
    template<class ...Args> decltype(auto) operator()(Args &&...args) { return fun_(std::ref(*this), std::forward<Args>(args)...); }
};
template<class Fun> decltype(auto) y_combinator(Fun &&fun) { return y_combinator_result<std::decay_t<Fun>>(std::forward<Fun>(fun)); }
#ifdef I_AM_NOOB
#define gg(...) [](const auto&...x){ char c='='; cerr<<#__VA_ARGS__; ((cerr<<exchange(c,',')<<x),...); cerr<<endl; }(__VA_ARGS__);
#else
#define gg(...) 777771449
#endif

constexpr ll INFF = 1e18;
constexpr ll P = 1e9+7;
// constexpr ll P = 998244353;

int main() {
    cin.tie(0) -> ios::sync_with_stdio(0);
    
    ll n; cin >> n;
    vector<string> a(n); for(auto&x:a) cin >> x;
    map<string, set<string>> eq;
    map<string, ll> mp;
    bool ok = true;
    f0(i, n) {
        string x; cin >> x;
        auto get = [&](string s) -> ll {
            return isdigit(s[0]) ? stoi(s) : mp[s];
        };
        auto assign = y_combinator([&](auto rec, string k, ll v) -> void {
            mp[k] = v;
            for (auto k1 : eq[k]) {
                rec(k1, v);
            }
            while (!eq[k].empty()) eq[k].erase(prev(eq[k].end()));
        });

        if (x == a[i]);
        else if (get(a[i]) && get(x)) {
            if (get(x) != get(a[i])) {
                cout << "NE" << endl;
                ok = false;
                break;
            }
        } else if (get(x)) {
            assign(a[i], get(x));
        } else if (get(a[i])) {
            assign(x, get(a[i]));
        } else {
            eq[a[i]].insert(x); eq[x].insert(a[i]);
        }
    }

    if (ok) {
        cout << "DA" << endl;
    }
    
    return 0;
}

• Subtask 1: 14 / 14
• Subtask 2: 0 / 14
• Subtask 3: 0 / 14
• Subtask 4: 0 / 14
• Subtask 5: 0 / 14