Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- public class HelloWorld{
- public static void main(String []args){
- int[] A = {3, 2, 4, 3};
- int F = 2;
- int M = 4;
- int[] ans = solution(A, F, M);
- print(ans);
- }
- public static int[] solution(int[] A, int F, int M) {
- int N = A.length;
- int[] ans = new int[F];
- int total = N + F;
- int unknown_sum = M * (N + F) - sum(A);
- int temp_unknown = unknown_sum;
- for(int i = 6, ans_pt = 0; ans_pt < F && i > 0; --i) {
- for(; (temp_unknown - i) >= (F - ans_pt - 1) && ans_pt < F; ++ans_pt) {
- ans[ans_pt] = i;
- temp_unknown -= i;
- }
- }
- if (sum(ans) != unknown_sum) {
- int[] x = {0};
- return x;
- }
- return ans;
- }
- public static int sum(int[] arr) {
- int sum = 0;
- for (int i = 0; i < arr.length; i++)
- sum+=arr[i];
- return sum;
- }
- public static void print(int[] arr) {
- for (int i = 0; i < arr.length; i++){
- System.out.println(arr[i]);
- }
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement