Submission #146346

#	Time	Username	Problem	Language	Result	Execution time	Memory
146346		jacynkaa	Traffic (CEOI11_tra)	C++14	100 / 100	1574 ms	96044 KiB

tra

#include <bits/stdc++.h>
#include <math.h>
#include <chrono>
using namespace std;
#pragma GCC optimize("-O3")
#define endl "\n"
#define mp make_pair
#define st first
#define nd second
#define pii pair<int, int>
#define pb push_back
#define _upgrade ios_base::sync_with_stdio(0), cout.setf(ios::fixed), cout.precision(10) //cin.tie(0); cout.tie(0);
#define REP(i, n) for (int i = 0; i < (n); ++i)
#define FWD(i, a, b) for (int i = (a); i < (b); ++i)
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define fwd(i, a, b) for (int i = (a); i < (b); ++i)
#define all(c) (c).begin(), (c).end()
#define what(x) cerr << #x << " is " << x << endl;

struct SilnieSpojne
{
    static const int MMM = 1e6 + 100; //zmienic jezeli potrzeba
    int n;
    vector<vector<int>> G, GT, Gspojne; //graf, graf z odwroconymi krawedziami, graf na spojnych
    vector<vector<int>> sklad;          //sklad kazdej ze spojnych
    vector<int> doKtorej;               // do kotrej spojnej nalezy wierzcholek
    bitset<MMM> odwiedzone;             //pomocnicze
    stack<int> kolejnosc;

    void dfs1(int x)
    {
        odwiedzone[x] = true;
        for (int y : G[x])
            if (!odwiedzone[y])
                dfs1(y);
        kolejnosc.push(x);
    }

    void dfs2(int x, int ktoraSpojna)
    {
        doKtorej[x] = ktoraSpojna;
        sklad[ktoraSpojna].pb(x);
        for (int y : GT[x])
            if (doKtorej[y] == -1)
                dfs2(y, ktoraSpojna);
    }

    inline int ileSpojnych()
    {
        return Gspojne.size();
    }

    void ini(vector<vector<int>> &G) //zakladamy numeracje wierzcholkow od 0, choc dla 1 tez (chyba) dziala
    {
        this->G = G;
        n = G.size();
        GT.resize(n);
        doKtorej.resize(n, -1);
        rep(i, n) for (int a : G[i]) GT[a].push_back(i); //tworzenie odwroconego grafu

        rep(i, n) if (!odwiedzone[i]) dfs1(i);
        while (!kolejnosc.empty())
        {
            int x = kolejnosc.top();
            kolejnosc.pop();
            if (doKtorej[x] == -1)
            {
                sklad.pb(vector<int>());
                dfs2(x, sklad.size() - 1);
            }
        }

        Gspojne.resize(sklad.size());
        vector<int> P(ileSpojnych(), -1);                                                                            //pomocniczy zaznaczamy czy juz mamy krawedz do danej spojnej
        rep(i, ileSpojnych()) for (int x : sklad[i]) for (int y : G[x]) if (doKtorej[y] != i && P[doKtorej[y]] != i) //dorzucam krawedzi do grafu
        {
            Gspojne[i].pb(doKtorej[y]);
            P[doKtorej[y]] = i;
        }
    }
    void debug()
    {
        cerr << "SKLADY SPOJNYCH:" << endl;
        rep(i, ileSpojnych())
        {
            auto A = sklad[i];
            cerr << "sklad " << i << " to: ";
            for (auto b : A)
                cerr << b + 1 << " ";
            cerr << endl;
        }
        cerr << "GRAF NA SPOJNYCH:" << endl;
        rep(i, ileSpojnych())
        {
            cerr << "sasiedzi wierzchołka " << i << ": ";
            for (auto b : Gspojne[i])
                cerr << b << " ";
            cerr << endl;
        }
    }
};

const int MAXN = 4e5;
int A, B, n, m;
vector<vector<int>> G;
pii punkty[MAXN];
pii val[MAXN];
vector<int> koncowe, poczatkowe; //posortowane po gorze
bitset<MAXN> visited;
SilnieSpojne S;

void wczytywanie()
{
    cin >> n >> m >> A >> B;
    G.resize(n);
    rep(i, n) cin >> punkty[i].st >> punkty[i].nd;

    rep(i, m)
    {
        int a, b, c;
        cin >> a >> b >> c;
        a--;
        b--;
        G[a].pb(b);
        if (c == 2)
            G[b].pb(a);
    }
}

void dfs1(int x)
{
    visited[x] = true;
    for (int y : G[x])
        if (!visited[y])
            dfs1(y);
}

void pre()
{
    rep(i, n) if (punkty[i].st == 0) poczatkowe.pb(i);
    for (int x : poczatkowe)
        if (!visited[x])
            dfs1(x);

    rep(i, n) if (punkty[i].st == A && visited[i]) koncowe.pb(i);

    sort(all(poczatkowe), [](int a, int b) { return punkty[a].nd > punkty[b].nd; });
    sort(all(koncowe), [](int a, int b) { return punkty[a].nd > punkty[b].nd; });
    S.ini(G);

    fill(val, val + MAXN, mp(MAXN, -MAXN)); //UWAGA
}

void merge(pii &A, pii B)
{
    A = mp(min(A.st, B.st), max(A.nd, B.nd));
}

void dfs2(int x)
{
    visited[x] = true;
    for (int y : S.Gspojne[x])
        if (!visited[y])
            dfs2(y);
    for (int y : S.Gspojne[x])
        merge(val[x], val[y]);
}

void solve()
{
    rep(i, koncowe.size())
        merge(val[S.doKtorej[koncowe[i]]], {i, i});
    //rep(i, S.ileSpojnych()) cerr << "wartosc wierzcholka : " << i << " to: " << val[i].st << " " << val[i].nd << endl;
    visited.reset();
    rep(i, S.ileSpojnych()) dfs2(i);

    for (int a : poczatkowe)
        cout << max(0, val[S.doKtorej[a]].nd - val[S.doKtorej[a]].st + 1) << endl;
}

void debug()
{
    S.debug();
    cerr << "POCZATKOWE" << endl;
    for (int a : poczatkowe)
        cout << a + 1 << " ";
    cout << endl;
    cerr << "KONCOWE" << endl;
    for (int a : koncowe)
        cout << a + 1 << " ";
    cout << endl;

    rep(i, S.ileSpojnych()) cerr << "wartosc wierzcholka : " << i << " to: " << val[i].st << " " << val[i].nd << endl;
}

main()
{
    _upgrade;
    wczytywanie();
    pre();
    solve();
    //debug();
}

Compilation message (stderr)

tra.cpp: In function 'void solve()':
tra.cpp:15:37: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i, n) for (int i = 0; i < (n); ++i)
                                     ^
tra.cpp:171:5: note: in expansion of macro 'rep'
     rep(i, koncowe.size())
     ^~~
tra.cpp: At global scope:
tra.cpp:196:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main()
      ^

• Subtask 1: 8 / 8
• Subtask 2: 8 / 8
• Subtask 3: 8 / 8
• Subtask 4: 8 / 8
• Subtask 5: 8 / 8
• Subtask 6: 8 / 8
• Subtask 7: 8 / 8
• Subtask 8: 8 / 8
• Subtask 9: 9 / 9
• Subtask 10: 9 / 9
• Subtask 11: 9 / 9
• Subtask 12: 9 / 9