Submission #522662

#	Time	Username	Problem	Language	Result	Execution time	Memory
522662		FronPaw	KOVANICE (COI15_kovanice)	C++14	0 / 100	283 ms	27324 KiB

kovanice

#include <bits/stdc++.h>
using namespace std;
int lg, n, m;
int unde[300001];
struct T
{
    int nr1, nr2;
    char cmp;
};
T op[300001];
int tata[300001], rang[300001];
int in[300001], in1[300001];
vector <int> v[300001];
bool viz[300001];
int ans[300001];
int find_tata (int nod)
{
    int rez = nod;
    while (nod != tata[nod])
        nod = tata[nod];
    while (rez != nod)
    {
        int cop = tata[rez];
        tata[rez] = nod;
        rez = cop;
    }
    return nod;
}
void uneste (int a, int b)
{
    if (rang[a] > rang[b])
        tata[b] = a;
    else
    {
        tata[a] = b;
        if (rang[a] == rang[b])
            rang[a]++;
    }
}
void solve ()
{
    queue <int> q ;
    vector <int> ordine;
    for (int i = 1; i<=n; ++i)
        if (in[unde[i]] == 0 && viz[unde[i]] == 0)
        {
            q.push(unde[i]);
            viz[unde[i]] = 1;
        }
    while (!q.empty())
    {
        int u = q.front();
        q.pop();
        ordine.push_back(u);

        for (auto vecin:v[u])
        {
            in[vecin] --;
            if (in[vecin] == 0)
                q.push(vecin);
        }
    }
    for (auto i:ordine)
        cout << in1[i] << ' ';
    cout << '\n';
    int current_sum = 1;
    for (int i = 1; i<ordine.size(); ++i)
    {
        if (in1[ordine[i]] == 0)
        {
            if (current_sum == lg)
                for (int j = i - 1, rr = 1; rr <= lg; --j, ++rr)
                    ans[ordine[j]] = rr;
            else
                for (int j = i - 1, rr = 1; rr <= current_sum ; --j, ++rr)
                    ans[ordine[j]] = -1;
            current_sum = 1;
        }
        else
            current_sum++;
    }
    if (current_sum == lg)
        for (int j = ordine.size() - 1, rr = 1; rr <= lg ; j--, ++rr)
            ans[ordine[j]] = rr;
    else
        for (int j = ordine.size() - 1, rr = 1; rr <= current_sum; j--, ++rr)
            ans[ordine[j]] = -1;
    return ;
}
int main ()
{
    cin >> lg >> n >> m;
    for (int i = 1; i<=m; ++i)
    {
        int a, b;
        char c;
        cin >> a >> c >> b;
        op[i].nr1 = a;
        op[i].nr2 = b;
        op[i].cmp = c;
    }
    for (int i = 1; i<=n; ++i)
        tata[i] = i;
    for (int i = 1; i<=m; ++i)
        if (op[i].cmp == '=')
            uneste(find_tata(op[i].nr1), find_tata(op[i].nr2));
    for (int i = 1; i<=n; ++i)
        unde[i] = find_tata(i);
    // use unde[nod] instead of nod
    for (int i = 1; i<=m; ++i)
        if (op[i].cmp == '<')
        {
            v[unde[op[i].nr2]].push_back(unde[op[i].nr1]);
            in[unde[op[i].nr1]]++;
            in1[unde[op[i].nr1]]++;
        }
        else if (op[i].cmp == '>')
        {
            v[unde[op[i].nr1]].push_back(unde[op[i].nr2]);
            in[unde[op[i].nr2]]++;
            in1[unde[op[i].nr2]]++;
        }
    solve();

    for (int i = 1;i<=n;++i)
        if (ans[unde[i]] == -1)
            cout << "?\n";
        else
            cout << "K" <<  ans[unde[i]] << '\n';
    return 0;
}

Compilation message (stderr)

kovanice.cpp: In function 'void solve()':
kovanice.cpp:67:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   67 |     for (int i = 1; i<ordine.size(); ++i)
      |                     ~^~~~~~~~~~~~~~

• Subtask 1: 0 / 10
• Subtask 2: 0 / 40
• Subtask 3: 0 / 10
• Subtask 4: 0 / 40