Submission #1019471

#	Time	Username	Problem	Language	Result	Execution time	Memory
1019471		stefanopulos	Dijamant (COI16_dijament)	C++17	100 / 100	499 ms	9304 KiB

dijament

#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <bits/stdc++.h>
 
using namespace std;
using namespace __gnu_pbds;
 
typedef long long ll;
typedef long double ldb;
 
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef pair<ldb,ldb> pdd;

#define ff(i,a,b) for(int i = a; i <= b; i++)
#define fb(i,b,a) for(int i = b; i >= a; i--)
#define trav(a,x) for(auto& a : x)
 
#define sz(a) (int)(a).size()
#define fi first
#define se second
#define pb push_back
#define lb lower_bound
#define ub upper_bound
#define all(a) a.begin(), a.end()
#define rall(a) a.rbegin(), a.rend()
 
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

template<typename T>
using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;

// os.order_of_key(k) the number of elements in the os less than k
// *os.find_by_order(k)  print the k-th smallest number in os(0-based)

const int mod = 1000000007;
const int inf = 1e9 + 5;
const int mxN = 2005; 

int n;

vector<int> g[mxN];

bool P[mxN][mxN];

bool visited[mxN];
void dfs(int v, int r){
    visited[v] = 1; P[r][v] = 1;
    for(auto u : g[v]){
        if(!visited[u]){
            dfs(u, r);
        }
    }
}


int main(){
    cin.tie(0)->sync_with_stdio(0);

    cin >> n; 

    map<string,int> mp; vector<int> A;
    ff(i,1,n){
        
        string X; cin >> X;
        bool ok = (mp.count(X) == 0); 

        string s; cin >> s; vector<int> vec;
        while(1){
            cin >> s;
            if(s == ";")break;
            if(mp.count(s))vec.pb(mp[s]);
            else ok = 0;
        }

        if(ok == 0){
            cout << "greska" << '\n';
            continue;
        }

        for(auto c : vec)g[i].pb(c);
        ff(j,1,n)visited[j] = 0;
        dfs(i, i);

        for(auto a : A){
            if(!P[i][a])continue;

            for(auto b : A){
                if(!P[i][b] || P[a][b] || P[b][a])continue;

                ff(c,1,n){
                    if(P[a][c] && P[b][c]){
                        ok = 0;
                        break;
                    }
                }

                if(ok == 0)break;

            }
            if(ok == 0)break;

        }

        if(ok == 0){
            for(auto c : vec)g[i].pop_back();
            ff(j,1,n)P[i][j] = 0;
        }
        else mp[X] = i, A.pb(i);

        cout << (ok == 0 ? "greska" : "ok") << '\n';
    }

    return 0;
}
/*

13
1 : ;
2 : 1 ;
3 : 1 ;
4 : 2 ;
5 : 3 ;
6 : 4 5 ;
7 : 4 5 ;
8 : 4 5 ;
9 : 4 5 ;
10 : 2 3 ;
11 : 2 4 ;
12 : 4 3 :
13 : 2 1 :


// probati bojenje sahovski
*/
 
 
 
 
 

Compilation message (stderr)

dijament.cpp: In function 'int main()':
dijament.cpp:106:22: warning: unused variable 'c' [-Wunused-variable]
  106 |             for(auto c : vec)g[i].pop_back();
      |                      ^

• Subtask 1: 13 / 13
• Subtask 2: 14 / 14
• Subtask 3: 29 / 29
• Subtask 4: 44 / 44