Samkit5025

Untitled

Jun 13th, 2022
55
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.42 KB | None | 0 0
  1. import java.util.*;
  2.  
  3. public class Solution {
  4.  
  5. static Scanner sc = new Scanner(System.in);
  6.  
  7. public static int[] completeAssignment(int N,int[] A){
  8. boolean[] prime = new boolean[100005];
  9. for(int i=0;i<100005;i++) {
  10. prime[i] = true;
  11. }
  12.  
  13. for(int i=2;i*i<=100005;i++) {
  14. if(prime[i] == true) {
  15. for(int j= i*i;j<100005;j+=i) {
  16. prime[j] = false;
  17. }
  18. }
  19. }
  20.  
  21. List<Integer> primeMember = new ArrayList<Integer>();
  22. List<Integer> nonPrimeMember = new ArrayList<Integer>();
  23.  
  24. for(int i : A) {
  25. if(prime[i] == true) {
  26. primeMember.add(i);
  27. }
  28. else {
  29. nonPrimeMember.add(i);
  30. }
  31. }
  32.  
  33. Collections.sort(primeMember);
  34. Collections.sort(nonPrimeMember);
  35.  
  36. int[] result = new int[N];
  37.  
  38. int start = 0;
  39.  
  40. for(int i : primeMember) {
  41. result[start] = i;
  42. start++;
  43. }
  44.  
  45. for(int i : nonPrimeMember) {
  46. result[start] = i;
  47. start++;
  48. }
  49.  
  50. return result;
  51. }
  52.  
  53. public static void main(String[] args) {
  54. int N = sc.nextInt();
  55. int[] A = new int[N];
  56.  
  57. for(int i=0;i<N;i++) {
  58. A[i] = sc.nextInt();
  59. }
  60.  
  61. int[] result = completeAssignment(N,A);
  62.  
  63. for(int i=0;i<N;i++) {
  64. System.out.print(result[i] + " ");
  65. }
  66. System.out.println();
  67. }
  68. }
Advertisement
Add Comment
Please, Sign In to add comment