Submission #560175

#	Time	Username	Problem	Language	Result	Execution time	Memory
560175		AGE	Zamjena (COCI18_zamjena)	C++14	70 / 70	639 ms	12936 KiB

zamjena

#include<bits/stdc++.h>
#define F first
#define S second
#define pb push_back
#define int long long
using namespace std;
const int N=6e4,M=2e3+10,mod=100000000000031;
vector<int>v;

long long  summ(long long  x,long long  y){
    return ((x%mod)+(y%mod))%mod;
}

long long  mult(long long  x,long long  y){
    return ((x%mod)*(y%mod))%mod;
}
string a[N],b[N];
map< string ,string > p;
map<string,int>sz;

string root(string i)
{
    if(p[i]!=p[p[i]])
        p[i]=root(p[i]);

    return p[i];
}
void onion(string x,string y){

    x=root(x);
    y=root(y);
    if(x==y)
        return ;

    if(sz[x]<sz[y])
        swap(x,y);

    p[y]=x;
    sz[x]+=sz[y];

}

bool okk(string s){
    return (s[0]>='a'&&s[0]<='z');
}


main()
{

    int n;
    cin>>n;

    for(int i=0;i<n;i++)
        cin>>a[i];

    for(int i=0;i<n;i++)
        cin>>b[i];

    for(int i=0;i<n;i++)
        p[a[i]]=a[i],p[b[i]]=b[i],sz[a[i]]=0,sz[b[i]]=0;

    int ok=0;

    for(int i=0;i<n;i++)
        if(okk(a[i])&&okk(b[i])){
            onion(a[i],b[i]);
        }

    for(int i=0;i<n;i++){

        if(!okk(a[i]))
            sz[a[i]]=100001;

        if(!okk(b[i]))
            sz[b[i]]=100001;

    }

    for(int i=0;i<n;i++){

        if(!okk(a[i])&&okk(b[i]))
            onion(b[i],a[i]);

        else if(okk(a[i])&&!okk(b[i]))
            onion(b[i],a[i]);

    }


    for(int i=0;i<n;i++){

        if(okk(a[i])){

            string x=root(a[i]);
            if(!okk(x))
                a[i]=x;
            else a[i]="1";

        }

        if(okk(b[i])){

            string x=root(b[i]);
            if(!okk(x))
                b[i]=x;
            else b[i]="1";

        }

    }

    for(int i=0;i<n;i++)
        if(a[i]!=b[i])
            ok=1;


    if(ok==1)
        cout<<"NE"<<endl;

    else cout<<"DA"<<endl;

    return 0;

}
/*
4
4 5 iks ipsilon
1 iks 3 iks

5
x 3 x y 3
x y 2 z 3

*/

Compilation message (stderr)

zamjena.cpp:48:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   48 | main()
      | ^~~~

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