Submission #846270

#	Time	Username	Problem	Language	Result	Execution time	Memory
846270		vjudge1	KOVANICE (COI15_kovanice)	C++17	100 / 100	485 ms	52424 KiB

kovanice

#pragma GCC optimize("unroll-loops,Ofast,O3")
#include <bits/stdc++.h>
#define pb push_back
#define mp make_pair
#define spc << " " <<
#define all(x) x.begin(), x.end()
#define ll long long
#define int long long
#define ii pair<int,int>
#define vi vector<int>
#define vii vector<ii>
#define st first
#define nd second
#define inf 1000000009
#define MOD 998244353
using namespace std;



int dad[300005];
string ans[300005];
int dep[300005];
vi edges[300005];

int find(int x){
    if(dad[x]==x) return x;
    dad[x]=find(dad[x]);
    return dad[x];
}
void unite(int x, int y){
    dad[find(x)]=find(y);
}


void dfs(int node){
    if(dep[find(node)]!=-1) return;
    dep[find(node)]=1;

    for(auto i:edges[find(node)]){
        dfs(find(i));
        dep[find(node)]=max(dep[find(node)], dep[find(i)]+1);
    }
}


void mark(int node){
    if(ans[find(node)]!="?") return;

    ans[find(node)] = "K";
    ans[find(node)] += to_string(dep[find(node)]);

    for(auto i:edges[find(node)]){
        if(dep[find(i)]==dep[find(node)]-1){
            mark(find(i));
        }
    }
}





void solve(){
    int n, m, v; cin >> n >> m >> v;

    for(int i=1; i<=m; i++){
        dad[i]=i;
        ans[i]="?";
        dep[i]=-1;
    }


    vector<string> later;
    for(int i=1; i<=v; i++){
        string s; cin >> s;
        int a=0, b=0, ptr=0;
        while(s[ptr]!='=' && s[ptr]!='<'){
            a*=10;
            a+=s[ptr]-'0';
            ptr++;
        }
        char comp=s[ptr];
        ptr++;
        while(ptr<s.size()){
            b*=10;
            b+=s[ptr]-'0';
            ptr++;
        }
        if(comp=='='){
            unite(find(a),find(b));
        }
        else later.pb(s);
    }
    for(auto s:later){
        int a=0, b=0, ptr=0;
        while(s[ptr]!='=' && s[ptr]!='<'){
            a*=10;
            a+=s[ptr]-'0';
            ptr++;
        }
        char comp=s[ptr];
        ptr++;
        while(ptr<s.size()){
            b*=10;
            b+=s[ptr]-'0';
            ptr++;
        }
        edges[find(b)].pb(find(a));
    }

    
    for(int i=1; i<=m; i++){
        dfs(find(i));
    }

    for(int i=1; i<=m; i++){
        if(dep[find(i)]==-1) cerr << "wtf" << endl;
        if(dep[find(i)]==n) mark(find(i));
    }

    for(int i=1; i<=m; i++){
        cout << ans[find(i)] << endl;
    }
}



signed main(){
    ios_base::sync_with_stdio(false);cin.tie(0);
    #ifdef Local
    freopen("in.txt","r",stdin);
    freopen("out","w",stdout);
    #endif


    ll t=1;
    //cin >> t;
    while(t--) solve();
}

Compilation message (stderr)

kovanice.cpp: In function 'void solve()':
kovanice.cpp:84:18: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   84 |         while(ptr<s.size()){
      |               ~~~^~~~~~~~~
kovanice.cpp:103:18: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  103 |         while(ptr<s.size()){
      |               ~~~^~~~~~~~~
kovanice.cpp:101:14: warning: unused variable 'comp' [-Wunused-variable]
  101 |         char comp=s[ptr];
      |              ^~~~

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