Kirjat

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Scanner;

public class Ohjelma {
    public static void main(String[] args) {
        Scanner io = new Scanner(System.in);
        int books = Integer.parseInt(io.nextLine());
        
        ArrayList<Integer> uo = new ArrayList<>();
        String un = io.nextLine();
        String temp = "";
        for (int i = 0; i < un.length(); i++) {
            if (un.charAt(i)==' ') {
                uo.add(Integer.parseInt(temp));
                temp = "";
            } else {
                temp+=un.charAt(i);
            }
            
        }
        uo.add(Integer.parseInt(temp));
        temp = "";
        ArrayList<Integer> ma = new ArrayList<>();
        String mas = io.nextLine();
        for (int i = 0; i < mas.length(); i++) {
            if (mas.charAt(i)==' ') {
                ma.add(Integer.parseInt(temp));
                temp = "";
            } else {
                temp+=mas.charAt(i);
            }
            
        }
        ma.add(Integer.parseInt(temp));
        
        ArrayList<Integer> end = new ArrayList<>();
        for (int i = 0; i < books; i++) {
            end.add(0);
        }
        ArrayList<Integer> tempo = new ArrayList<>();
        ArrayList<Integer> tempo2 = new ArrayList<>();
        for (int i = 0; i < books; i++) {
            Boolean wasAdded = false;
            
            for (int j = 0; j < books; j++) {
                
                if (i+1 != ma.get(j) && i+1 != uo.get(j) && end.get(j)==0) {
                    end.set(j, i+1);
                    wasAdded = true;
                    break;
                }
                
            }
            if (!wasAdded) {
                tempo.add(i+1);
              
            }
            
        }
        for (int i = 0; i < end.size(); i++) {
            if (end.get(i)==0) {
                tempo2.add(i);
            }
        }
        //tempo2 includes the positions of 0 ints
        //tempo includes the values not added
        
        for (int i = 0; i < tempo.size(); i++) {
            for (int j = 0; j < books; j++) {
                if ((tempo.get(i) != ma.get(j) && tempo.get(i) != uo.get(j)) 
                        && ma.get(j) != tempo.get(i) && uo.get(j) != tempo.get(i)) {
                    HashMap<Integer,Integer> tem = new HashMap<>();
                    tem.put(j, end.get(j));
                    end.set(j,tempo.get(i));
                    end.set(tempo2.get(i),tem.get(j));
                    tem.clear();
                    break;
                }
            }
            
        }
        
       /* for (int i : tempo2) {
            for (int j = 0; j < books; j++) {
                if (!(i+1 == ma.get(j) | i+1 == uo.get(j))) {
                    
                }
            }
        } */
        if (tempo.size()>1) {
            int nedd = 0/0;
        }
        for (int i = 0; i < end.size(); i++) {
            System.out.print(end.get(i) + " ");
        }
    }
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: WRONG ANSWER

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 1

Verdict: ACCEPTED

Test 7

Group: 1

Verdict: ACCEPTED

Test 8

Group: 1

Verdict: WRONG ANSWER

Test 9

Group: 1

Verdict: WRONG ANSWER

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: WRONG ANSWER

Test 24

Group: 2

Verdict: WRONG ANSWER

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: WRONG ANSWER

Test 32

Group: 2

Verdict: ACCEPTED

Test 33

Group: 2

Verdict: ACCEPTED

Test 34

Group: 2

Verdict: WRONG ANSWER

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: ACCEPTED

Test 45

Group: 3

Verdict: WRONG ANSWER

Test 46

Group: 3

Verdict: ACCEPTED

Test 47

Group: 3

Verdict: WRONG ANSWER

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: ACCEPTED

Test 54

Group: 3

Verdict: ACCEPTED

Test 55

Group: 3

Verdict: ACCEPTED

Test 56

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 57

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 58

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 59

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 60

Group: 3

Verdict: TIME LIMIT EXCEEDED