Ayush1325

Bhai Problem

Jun 20th, 2021
530
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. public class HelloWorld{
  2.      public static void main(String []args){
  3.         int[] A = {3, 2, 4, 3};
  4.         int F = 2;
  5.         int M = 4;
  6.         int[] ans = solution(A, F, M);
  7.        
  8.         print(ans);
  9.      }
  10.      
  11.      public static int[] solution(int[] A, int F, int M) {
  12.          int N = A.length;
  13.          int[] ans = new int[F];
  14.          int total = N + F;
  15.          
  16.          int unknown_sum = M * (N + F) - sum(A);
  17.          
  18.          int temp_unknown = unknown_sum;
  19.          
  20.          for(int i = 6, ans_pt = 0; ans_pt < F && i > 0; --i) {
  21.              for(; (temp_unknown - i) >= (F - ans_pt - 1) && ans_pt < F; ++ans_pt) {
  22.                  ans[ans_pt] = i;
  23.                  temp_unknown -= i;
  24.              }
  25.          }
  26.          
  27.          if (sum(ans) != unknown_sum) {
  28.              int[] x = {0};
  29.              return x;
  30.          }
  31.          
  32.          return ans;
  33.      }
  34.      
  35.      public static int sum(int[] arr) {
  36.         int sum = 0;
  37.          
  38.         for (int i = 0; i < arr.length; i++)
  39.             sum+=arr[i];
  40.         return sum;
  41.     }
  42.    
  43.     public static void print(int[] arr) {
  44.         for (int i = 0; i < arr.length; i++){
  45.             System.out.println(arr[i]);
  46.         }    
  47.     }
  48. }
RAW Paste Data