You are not logged in. Please login at to post your questions!


DISHOWN - Editorial




Author: Vivek Hamirwasia
Tester: Shang Jingbo and Gerald Agapov
Editorialist: Devendra Agarwal




Set Union Data Structure


N chef's are there. Each chef has a dish which is given a value (denoted by S[i]). You need to answer 2 queries :
1. 0 x y : Chef containing dish x competes with chef containing dish y. If both dishes belong to same chef print "Invalid Query!" , otherwise the winner of the round will obtain all the dishes of the loser which will then be eliminated.
2. 1 x : Output the chef which contains the dish x.


This problem can be solved using Disjoint set data structures.

Union Find or Disjoint Set Data structures allow you to merge two sets of items together dynamically and maintain for each item - to which set set does it belong.

This is exactly what we need in the problem , i.e. dynamically merging two sets and querying which set does an element belong.

We will be using Disjoint Set Data Structure that supports the following :

  1. find(A) : Get the id of the disjoint set to which A belongs(i.e. the chef id which contains that dish).
  2. union(A,B) : Merge the two disjoint sets in which A and B belong respectively.

Initially, we will create N disjoint set , each set contains 1 dish and it's owner.
Let dish x be in setA and dish y be in setB.
If the dish contained by the owner of setA has score higher than the dish contained by the owner of setB , then we will merge setA and setB and set the owner of setA as the overall owner of the merged set.
If the dish contained by the owner of setB has score higher than the dish contained by the owner of setA , then we will merge setA and setB and set the owner of setB as the overall owner of the merged set.

Note : You can easily prove from this that the owner of the set has dish whose score is higher than all other dish of the set.
We will be using only Path Compression heuristics to solve the problem.

Pseudo Code

Initialize parent[i] = i  
Let S[i] denote the initial array.

int find(int i)
    int j
                return i
        //Path Compression Heuristics
        return j

set_union(int i,int j)
    int x1,y1
    //parent of both of them will be the one with the highest score
    else if ( S[x1] < S[y1])

    if(query == 0)
        Input x and y
        px = find(x)
        py = find(y)
        if(px == py)
            print "Invalid query!"
        Input x.
        print find(x)

AUTHOR'S and TESTER'S Solution:

Author's solution
Tester's solution

This question is marked "community wiki".

asked 14 Jul '14, 15:01

devuy11's gravatar image

4★devuy11 ♦♦
accept rate: 0%

edited 29 Jan, 16:41

admin's gravatar image

0★admin ♦♦


I have solved with the same approach but was getting TLE as i was using Console.WriteLine to print all the output in C# . When i used the StringBuilder to store the messages and output in the last it passed the test cases. I am not sure if we have to consider this small optimization also

(14 Jul '14, 15:22) dev85462★

I have used same approach but I am getting TLE please help me!

(14 Jul '14, 15:31) mtbar1312★

@mtbar131 : You didnt use path compression. That's what I could notice.

(14 Jul '14, 15:33) himanshujaju6★

@dev8546: Thanks for your comment. Now I understand the reason that I never got it AC. I couldn't think that Console.WriteLine() might be causing the problem. I kept on hitting my head thinking there might be a better optimized way to solve the problem and finally gave up on it after trying some micro optimizations :(

(14 Jul '14, 15:35) ig_quasar1★

Did the same in Java as dev8546 described. By printing System.out.println() it gets TLE. If i use a stringbuilder and output it at the end it passes. (I used both path compression and union by rank).

(14 Jul '14, 15:43) szilard4★

Just a quick check - Is this true with all languages or just the managed ones like C#/JAVA?

(14 Jul '14, 16:13) ig_quasar1★

author's solution is giving TLE. i think path compression technique is not working. i also used that method and got tle. pls check it.

(14 Jul '14, 23:56) imrajeshpal3★

this code is giving TLE i cant understand why.

(15 Jul '14, 19:48) abhishekjhaji2★

Authors' solution for this problem is wrong. It gives TLE because it doesn't use path compression.

(23 Jul '14, 02:12) nitinj5★

What is the need of using find() function again in set_union() when we are already passing set_union() the parents of the 2 dishes?

(03 Sep '14, 18:42) samdcoder2★
showing 5 of 10 show all

123next »

@kuruma thanks for this question i learnt something new.

first tiny mistake is the output of second query type you should print find(x)+1 as your numbers are 0-based .

secondly you have used "ios_base::sync_with_stdio(false)" that seems to cause your output to be buffered and is printed together only when the program terminates.

this would have been correct had you used puts/printf in output of 2nd query as well. since you used cout that output is not buffered and printed instantly while the output of all first queries is printed in the end.

So what you can do:

1.use fflush(stdout) after puts statement

2.use cout

3.NOT use ios_base::sync_with_stdio(false);

4.use puts/printf in the output of 2nd query as well


answered 20 Jul '14, 01:36

shubham12's gravatar image

accept rate: 13%

edited 20 Jul '14, 01:49

Thanks a lot @shubham12 , I switched puts to cout and used the +1 trick and got AC :D

Really, really thanks a lot :) And the seudo-code from editorial was also really well written :D Hope I can start doing this more often

(20 Jul '14, 01:48) kuruma2★

glad i could help :). though i had some incorrect understanding of the effect of "ios_base::sync_with_stdio(false)". i have to read it up.

(20 Jul '14, 01:51) shubham123★


can anyone tell why TLE:,squal



answered 14 Jul '14, 15:29

squal's gravatar image

accept rate: 14%


don't use System.out.println(), instead use StringBuilder to store answers and then in the end use System.out.println() to print StringBuilder object. I have modified your code it has got accepted. Link to ur code:,sousnake (see the accepted solution).

(14 Jul '14, 16:03) sousnake2★

@sousnake: thanks :) bad that i missed this because of System.out.println() :(

(14 Jul '14, 16:49) squal4★

admins...plz see my code .. ..used same approach ..stll TLE.!!..its really annoying ..


answered 14 Jul '14, 15:37

rachilies's gravatar image

accept rate: 11%


yeah ..the path compression heuristics...really had an opportunity to learn.. btw real thanx. :)

(14 Jul '14, 16:32) rachilies4★

Hai, I don't know whether I can post this or not ,Here are some unofficial test cases categorised into small,medium,large.Who are getting mainly WA but also RTE,TLE,... can also check.You may need not worry about the constraints.the values are guarenteed to be in the given constraints.


answered 14 Jul '14, 15:38

achaitanyasai's gravatar image

accept rate: 10%

i have tried all your test files for DISHOWN and got all correct output.. but NZEC in codechef ..

see the below comment /!\

(14 Jul '14, 22:04) aq_faridi5★

Can anybody tell me why i was getting Runtime Error(NZEC) in java (i have used the same union-find logic)??
here is the code =>

When i transformed the same logic in c++, it got accepted... here is ACed c++ code =>

Thanx in Advance :)


answered 14 Jul '14, 15:39

aq_faridi's gravatar image

accept rate: 0%

I think the problem was in your function owner. I have modified that function just a bit and it got accepted. Here's the link:

(14 Jul '14, 17:39) sousnake2★

what is the difference between these two functions?? :

function 1 : gave AC

static int owner(int x)
    if(p[x] == x)
        return x;
        int var = owner(p[x]);
        p[x] = var;
        return p[x];

function 2: gives NZEC

static int owner(int x)
    if(p[x] == x)
        return x;
        p[x] = owner(p[x]);
        return p[x];
(14 Jul '14, 19:19) aq_faridi5★

no I think the problem was that: (this was your original code) static int owner(int x) { if(p[x] == x) return x; else return p[x]=owner(p[x]); }

here may be the problem was with assignment operator (i am not sure how it works but it works differently in java and C++).

(15 Jul '14, 18:58) sousnake2★

I used UF data structure with path compression but still got TLE. Why would the program run into TLE when using System.out.println()? Can anybody explain that?


answered 14 Jul '14, 16:54

nastra's gravatar image

accept rate: 0%

System.out.println() is slow .. use BufferedWriter or BufferedOutputStream for fast IO see my code for reference :

(15 Jul '14, 21:37) aq_faridi5★

Can anyone tell what's wrong with my code?Getting WA. Code


answered 14 Jul '14, 17:14

itsmepsk's gravatar image

accept rate: 0%

anyone would tell wats wrng in my code .. its same as that of the editorial !!!!! the link of the solution is .. any help is seriously welcomed


answered 14 Jul '14, 19:18

meintoo's gravatar image

accept rate: 16% You code gives WA for the test case Mentioned on this link. Also i have commented in one line where i have done the edit. After this, use fast I/O and it should give AC.

(14 Jul '14, 23:53) damn_me3★

i don't know about how the test cases were designed but still if this algorithm is implemented with union by rank it will give better performance.I am a bit disappointed that my code that should have been accepted with better performance gave TLE if anybody can explain the reason for this do reply....


answered 14 Jul '14, 19:28

vaibhav9518s's gravatar image

accept rate: 0%

In Java BufferedReader is giving TLE.
Even taking only input without processing anything it gives TLE.Why this happened for java Language. It run smoothly for scanf in c if it fastio problem then it should give TLE for scanf also.
This is my code in which only input is getting take by program and for this it is giving TLE.


answered 14 Jul '14, 20:46

shivamsharma's gravatar image

accept rate: 0%

edited 14 Jul '14, 20:48


i think you should use BufferdOutputStream instead of System.out.println see my code for reference :

(15 Jul '14, 21:41) aq_faridi5★

@aq_faridi I took only input then also I got TLE.

(15 Jul '14, 22:06) shivamsharma3★
toggle preview

Follow this question

By Email:

Once you sign in you will be able to subscribe for any updates here



Answers and Comments

Markdown Basics

  • *italic* or _italic_
  • **bold** or __bold__
  • link:[text]( "title")
  • image?![alt text](/path/img.jpg "title")
  • numbered list: 1. Foo 2. Bar
  • to add a line break simply add two spaces to where you would like the new line to be.
  • basic HTML tags are also supported
  • mathemetical formulas in Latex between $ symbol

Question tags:


question asked: 14 Jul '14, 15:01

question was seen: 12,776 times

last updated: 25 Mar, 19:55