CSES twosets java

    import java.io.*;
import java.math.BigInteger;
import java.util.*;
import java.util.StringTokenizer;

    public class twosets {
        public static void main(String[] args) throws IOException {
            FastScanner in = new FastScanner();
            long n = in.nextInt();

            BigInteger tSum = BigInteger.valueOf(n * (n+1)/2);

            if( !tSum.mod(BigInteger.valueOf(2)).equals(BigInteger.valueOf(0))) {
            	System.out.println("NO");
            }
            else {
            	System.out.println("YES");
            	List<Integer> set1 = new ArrayList<>();
            	List<Integer> set2 = new ArrayList<>();
            	BigInteger target = BigInteger.valueOf((n*(n+1))/4);

            	BigInteger set1Sum = BigInteger.ZERO;

            	for (int i = 1; i <= n; i++) {
            		//System.out.println(target);
            		if (set1Sum.add(BigInteger.valueOf(i)).compareTo(target) < 0) {
            			set1Sum = set1Sum.add(BigInteger.valueOf(i));
            			set1.add(i);
            		}
            		else {
            			// find the largest number u can add to this while removing a num
            			// i is the current largest number we have

            			// to replace the number, find the difference between target and n = i, then replace the i'th term with n;

            			BigInteger dif = BigInteger.valueOf(n).subtract(target.subtract(set1Sum));
            			int diff = Integer.parseInt(dif.toString());
            			set1.remove(diff-1);
            			set1.add((int)n);
            			break;
            		}
            	}

            	int c = 0;
            	for (int i = 1; i <= n; i++) {
            		if (c < set1.size()) {
            			if (set1.get(c) != i) {
                			set2.add(i);
                		}
                		else c++;
            		}
            		else set2.add(i);
            	}



            	System.out.println(set1.size());
            	for (int x: set1) {
            		System.out.print(x + " ");
            	}
            	System.out.println();

            	System.out.println(set2.size());
            	for (int x: set2) {
            		System.out.print(x + " ");
            	}
            	System.out.println();
            }
        }

        public static class FastScanner {
            BufferedReader br;
            StringTokenizer st;

            public FastScanner(String s) {
                try {
                    br = new BufferedReader(new FileReader(s));
                } catch (FileNotFoundException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
            }

            public FastScanner() {
                br = new BufferedReader(new InputStreamReader(System.in));
            }

            String nextToken() {
                while (st == null || !st.hasMoreElements()) {
                    try {
                        st = new StringTokenizer(br.readLine());
                    } catch (IOException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                    }
                }
                return st.nextToken();
            }

            String nextLine() {
                st = null;
                try {
                    return br.readLine();
                } catch (IOException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                return null;
            }

            int nextInt() {
                return Integer.parseInt(nextToken());
            }

            long nextLong() {
                return Long.parseLong(nextToken());
            }

            double nextDouble() {
                return Double.parseDouble(nextToken());
            }
        }
    }