본문 바로가기
1 Day 1 Algorithms

[2019.02.08] Birthday Chocolate

by 곰돌찌 2019. 2. 8.

Problem


Lily has a chocolate bar that she wants to share it with Ron for his birthday. Each of the squares has an integer on it. She decides to share a contiguous segment of the bar selected such that the length of the segment matches Ron's birth month and the sum of the integers on the squares is equal to his birth day. You must determine how many ways she can divide the chocolate.

Consider the chocolate bar as an array of squares, . She wants to find segments summing to Ron's birth day,  with a length equalling his birth month, . In this case, there are two segments meeting her criteria:  and .

Function Description

Complete the birthday function in the editor below. It should return an integer denoting the number of ways Lily can divide the chocolate bar.

birthday has the following parameter(s):

  • s: an array of integers, the numbers on each of the squares of chocolate
  • d: an integer, Ron's birth day
  • m: an integer, Ron's birth month

Input Format

The first line contains an integer , the number of squares in the chocolate bar. 
The second line contains  space-separated integers , the numbers on the chocolate squares where 
The third line contains two space-separated integers,  and , Ron's birth day and his birth month.

Constraints

  • , where ()

Output Format

Print an integer denoting the total number of ways that Lily can portion her chocolate bar to share with Ron.

Sample Input 0

5
1 2 1 3 2
3 2

Sample Output 0

2

Explanation 0

Lily wants to give Ron  squares summing to . The following two segments meet the criteria:

image

Sample Input 1

6
1 1 1 1 1 1
3 2

Sample Output 1

0

Explanation 1

Lily only wants to give Ron  consecutive squares of chocolate whose integers sum to . There are no possible pieces satisfying these constraints:

image

Thus, we print  as our answer.

Sample Input 2

1
4
4 1

Sample Output 2

1

Explanation 2

Lily only wants to give Ron  square of chocolate with an integer value of . Because the only square of chocolate in the bar satisfies this constraint, we print  as our answer.


How I solved the problem


# 초콜릿 바는 이어져있기 때문에 하나하나 분리해서 못함!

# 더해서 초콜릿을 m개씩 더해서 d가 되는지를 확인하면 되는 문제


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import java.io.*;
import java.math.*;
import java.security.*;
import java.text.*;
import java.util.*;
import java.util.concurrent.*;
import java.util.function.*;
import java.util.regex.*;
import java.util.stream.*;
 
import com.sun.swing.internal.plaf.metal.resources.metal_es;
 
import static java.util.stream.Collectors.joining;
import static java.util.stream.Collectors.toList;
 
public class Solution {
 
    // Complete the birthday function below.
    static int birthday(List<Integer> s, int d, int m) {
        int sCount = s.size();
        int result = 0;
 
        for (int i = 0; i < sCount; i++) {
            int sum = 0;
 
            if (+ m > sCount) { 
                //if i + m is larger than list s's length, break for loop
                break;
            }
 
            for (int j = i; j < i + m; j++) {
                sum += s.get(j); // list value is added (consecutive)
            }
 
            if (sum == d) {
                result++;
            }
        }
 
        return result;
    }
 
    public static void main(String[] args) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bufferedWriter = new BufferedWriter(new FileWriter(System.getenv("OUTPUT_PATH")));
 
        int n = Integer.parseInt(bufferedReader.readLine().trim());
 
        List<Integer> s = Stream.of(bufferedReader.readLine().replaceAll("\\s+$""").split(" "))
            .map(Integer::parseInt)
            .collect(toList());
 
        String[] dm = bufferedReader.readLine().replaceAll("\\s+$""").split(" ");
 
        int d = Integer.parseInt(dm[0]);
 
        int m = Integer.parseInt(dm[1]);
 
        int result = birthday(s, d, m);
 
        bufferedWriter.write(String.valueOf(result));
        bufferedWriter.newLine();
 
        bufferedReader.close();
        bufferedWriter.close();
    }
}
 
cs

[출처 : https://www.hackerrank.com ]

'1 Day 1 Algorithms' 카테고리의 다른 글

[2019.02.12] Migratory Birds  (0) 2019.02.12
[2019.02.11] Divisible Sum Pairs  (0) 2019.02.11
[2019.02.01] Breaking the Records  (0) 2019.02.01
[2019.01.31] Between Two Sets  (0) 2019.01.31
[2019.01.30] Kangaroo  (0) 2019.01.30

댓글