Kirjat

#include <iostream>
#include <string>

int main()
{
	std::ios_base::sync_with_stdio( false );

	unsigned long bookCount;
	std::string input1;
	std::string input2;

	std::cin >> bookCount;
	std::cin.get();
	std::getline( std::cin, input1 );
	std::getline( std::cin, input2 );

	unsigned long* sequence1 = new unsigned long[ bookCount ];
	unsigned long* sequence2 = new unsigned long[ bookCount ];
	unsigned long* sequence3 = new unsigned long[ bookCount ];

	unsigned long long index = 0;
	unsigned long long cursor = 0;

	for ( unsigned long long i = 0; i < input1.length(); ++i )
	{
		if ( input1[ i ] == ' ' )
		{
			sequence1[ index ] = atoi( input1.substr( cursor, i - cursor ).c_str() );
			index++;
			cursor = i + 1;
		}
	}

	sequence1[ index ] = atoi( input1.substr( cursor, input1.npos ).c_str() );
	index = 0;
	cursor = 0;

	for ( unsigned long long i = 0; i < input2.length(); ++i )
	{
		if ( input2[ i ] == ' ' )
		{
			sequence2[ index ] = atoi( input2.substr( cursor, i - cursor ).c_str() );
			index++;
			cursor = i + 1;
		}
	}

	sequence2[ index ] = atoi( input2.substr( cursor, input2.npos ).c_str() );

	unsigned long lowestUnread = 1;
	bool* booksRead = new bool[ bookCount ];

	for ( unsigned long i = 0; i < bookCount; ++i )
	{
		booksRead[ i ] = false;
	}

	for ( unsigned long i = 0; i < bookCount; ++i )
	{
		if ( lowestUnread != sequence1[ i ] && lowestUnread != sequence2[ i ] )
		{
			sequence3[ i ] = lowestUnread;
			booksRead[ lowestUnread - 1 ] = true;

			while ( lowestUnread < bookCount && booksRead[lowestUnread - 1] )
			{
				lowestUnread++;
			}			
		}
		else
		{
			unsigned long j = lowestUnread + 1;
			bool mustFlip = true;

			while ( j <= bookCount )
			{
				if ( j != sequence1[ i ] && j != sequence2[ i ] && !booksRead[ j - 1 ] )
				{
					sequence3[ i ] = j;
					booksRead[ j - 1 ] = true;

					if ( j == lowestUnread )
					{
						while ( booksRead[ lowestUnread - 1 ] )
						{
							lowestUnread++;
						}
					}

					mustFlip = false;
					break;
				}

				j++;
			}

			if ( mustFlip )
			{
				for ( unsigned long k = 0; k < bookCount; ++k )
				{
					if ( lowestUnread != sequence1[ k ] && lowestUnread != sequence2[ k ]  &&
						sequence3[ k ] != sequence1[ i ] && sequence3[ k ] != sequence2[ i ] )
					{
						sequence3[ i ] = sequence3[ k ];
						sequence3[ k ] = lowestUnread;
						booksRead[ lowestUnread - 1 ] = true;
						break;
					}
				}
			}
		}		
	}

	for ( unsigned long i = 0; i < bookCount; ++i )
	{
		if ( sequence3[ i ] == sequence1[ i ] || sequence3[ i ] == sequence2[ i ] )
		{
			for ( unsigned long j = 0; j < bookCount; ++j )
			{
				if ( j != i && sequence3[ i ] != sequence1[ j ] && sequence3[ i ] != sequence2[ j ] &&
					sequence3[ j ] != sequence1[ i ] && sequence3[ j ] != sequence2[ i ] )
				{
					unsigned long temp = sequence3[ i ];
					sequence3[ i ] = sequence3[ j ];
					sequence3[ j ] = temp;
					break;
				}
			}
		}
	}

	for ( unsigned long i = 0; i < bookCount; ++i )
	{
		std::cout << sequence3[ i ] << " ";
	}

	delete[] sequence1;
	delete[] sequence2;
	delete[] sequence3;
	delete[] booksRead;
	return 0;
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 1

Verdict: ACCEPTED

Test 7

Group: 1

Verdict: ACCEPTED

Test 8

Group: 1

Verdict: ACCEPTED

Test 9

Group: 1

Verdict: ACCEPTED

Test 10

Group: 1

Verdict: ACCEPTED

Test 11

Group: 1

Verdict: ACCEPTED

Test 12

Group: 1

Verdict: ACCEPTED

Test 13

Group: 1

Verdict: ACCEPTED

Test 14

Group: 1

Verdict: ACCEPTED

Test 15

Group: 1

Verdict: ACCEPTED

Test 16

Group: 1

Verdict: ACCEPTED

Test 17

Group: 1

Verdict: ACCEPTED

Test 18

Group: 1

Verdict: ACCEPTED

Test 19

Group: 1

Verdict: ACCEPTED

Test 20

Group: 1

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: ACCEPTED

Test 22

Group: 2

Verdict: ACCEPTED

Test 23

Group: 2

Verdict: ACCEPTED

Test 24

Group: 2

Verdict: ACCEPTED

Test 25

Group: 2

Verdict: ACCEPTED

Test 26

Group: 2

Verdict: ACCEPTED

Test 27

Group: 2

Verdict: ACCEPTED

Test 28

Group: 2

Verdict: ACCEPTED

Test 29

Group: 2

Verdict: ACCEPTED

Test 30

Group: 2

Verdict: ACCEPTED

Test 31

Group: 2

Verdict: ACCEPTED

Test 32

Group: 2

Verdict: ACCEPTED

Test 33

Group: 2

Verdict: ACCEPTED

Test 34

Group: 2

Verdict: ACCEPTED

Test 35

Group: 2

Verdict: ACCEPTED

Test 36

Group: 2

Verdict: ACCEPTED

Test 37

Group: 2

Verdict: ACCEPTED

Test 38

Group: 2

Verdict: ACCEPTED

Test 39

Group: 2

Verdict: ACCEPTED

Test 40

Group: 2

Verdict: ACCEPTED

Test 41

Group: 3

Verdict: ACCEPTED

Test 42

Group: 3

Verdict: ACCEPTED

Test 43

Group: 3

Verdict: ACCEPTED

Test 44

Group: 3

Verdict: WRONG ANSWER

Test 45

Group: 3

Verdict: ACCEPTED

Test 46

Group: 3

Verdict: ACCEPTED

Test 47

Group: 3

Verdict: ACCEPTED

Test 48

Group: 3

Verdict: ACCEPTED

Test 49

Group: 3

Verdict: ACCEPTED

Test 50

Group: 3

Verdict: ACCEPTED

Test 51

Group: 3

Verdict: ACCEPTED

Test 52

Group: 3

Verdict: ACCEPTED

Test 53

Group: 3

Verdict: WRONG ANSWER

Test 54

Group: 3

Verdict: ACCEPTED

Test 55

Group: 3

Verdict: ACCEPTED

Test 56

Group: 3

Verdict: ACCEPTED

Test 57

Group: 3

Verdict: ACCEPTED

Test 58

Group: 3

Verdict: ACCEPTED

Test 59

Group: 3

Verdict: ACCEPTED

Test 60

Group: 3

Verdict: ACCEPTED