Tasks Details
The submission is being evaluated.
The submission is being evaluated.
easy
1.
Fish
N voracious fish are moving along a river. Calculate how many fish are alive.
Task Score
100%
Correctness
100%
Performance
100%
You are given two non-empty arrays A and B consisting of N integers. Arrays A and B represent N voracious fish in a river, ordered downstream along the flow of the river.
The fish are numbered from 0 to N − 1. If P and Q are two fish and P < Q, then fish P is initially upstream of fish Q. Initially, each fish has a unique position.
Fish number P is represented by A[P] and B[P]. Array A contains the sizes of the fish. All its elements are unique. Array B contains the directions of the fish. It contains only 0s and/or 1s, where:
- 0 represents a fish flowing upstream,
- 1 represents a fish flowing downstream.
If two fish move in opposite directions and there are no other (living) fish between them, they will eventually meet each other. Then only one fish can stay alive − the larger fish eats the smaller one. More precisely, we say that two fish P and Q meet each other when P < Q, B[P] = 1 and B[Q] = 0, and there are no living fish between them. After they meet:
- If A[P] > A[Q] then P eats Q, and P will still be flowing downstream,
- If A[Q] > A[P] then Q eats P, and Q will still be flowing upstream.
We assume that all the fish are flowing at the same speed. That is, fish moving in the same direction never meet. The goal is to calculate the number of fish that will stay alive.
For example, consider arrays A and B such that:
A[0] = 4 B[0] = 0 A[1] = 3 B[1] = 1 A[2] = 2 B[2] = 0 A[3] = 1 B[3] = 0 A[4] = 5 B[4] = 0Initially all the fish are alive and all except fish number 1 are moving upstream. Fish number 1 meets fish number 2 and eats it, then it meets fish number 3 and eats it too. Finally, it meets fish number 4 and is eaten by it. The remaining two fish, number 0 and 4, never meet and therefore stay alive.
Write a function:
class Solution { public int solution(int[] A, int[] B); }
that, given two non-empty arrays A and B consisting of N integers, returns the number of fish that will stay alive.
For example, given the arrays shown above, the function should return 2, as explained above.
Write an efficient algorithm for the following assumptions:
- N is an integer within the range [1..100,000];
- each element of array A is an integer within the range [0..1,000,000,000];
- each element of array B is an integer that can have one of the following values: 0, 1;
- the elements of A are all distinct.
Copyright 2009–2025 by Codility Limited. All Rights Reserved. Unauthorized copying, publication or disclosure prohibited.
Solution
Programming language used Java 21
Time spent on task 52 minutes
Notes
not defined yet
Code: 06:23:42 UTC,
java,
verify,
result: Failed
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
// you can also use imports, for example:
// import java.util.*;
// you can use System.out.println for debugging purposes, e.g.
// System.out.println("this is a debug message");
class Solution {
public int solution(int[] A, int[] B) {
// write your code in Java SE 8
/*
문제:
입력되는 배열 A의 index는 물기고번호, 값은 물고기 크기다.
입력되는 배열 B의 index는 물고기번호, 값은 물고기가 향하는 방향이다.
배열 B의 값이 0이면 상류로 가는 것,
배열 B의 값이 1이면 하류로 가는 것이다.
index의 위치가 물고기가 처음 배치된 위치이다.
그렇기 때문에 위쪽에 위치한 물고기는 아래에 있는 물고기를 만날수 없다.
또한, 같은 방향으로 향하는 물고기도 만날 수 없다.
다른 방향으로 향하는 물고기를 만나면 크기가 큰 물고기가 작은 물고리를 잡아 먹는다.
방법:
1. index가 작을수록(P < Q)일수록 상류에 위치한다.
2. 하류에 위치한 물고기에 초점을 맞추고 자신보다 아래에서 올라오는 물고기와 비교한다.
3. 살아남은 물고기 들은 스택에 저장한다.
시나리오:
0. 물고기가 하류로 향하면 스택에 넣는다.
0-1. 다음 물고기가 상류로 향하는 물고기 일 경우 스택에서 빼와 비교한다.
0-2. 상류 물고기의 크기가 크다면 비교 물고기를 먹는다(없엔다)
0-3. 상류 물고기가 작다면 스택에 그대로 있다.
1. 물고기가 상류로 향하고, 비교 물고기(하류로향하는 물고기)가 없으면 살아남은 물고기
2. 물고기가 하류로 향하고, 비교 물고기가 있으면, 비교 물고기는 하류 스택에 넣는다.
*/
int living_count = 0;
Stack downstream = new Stack<Integer>();
for(int i = 0; i<A.length; i++){
int stream = B[i];
if(B[i] == 0){ // 물고기가 상류로 향하면
// 1. 하류로 향하는 물고기가 있으면 비교, 없으면 살아서 올라감.
if(!downstream.empty()){
int eat_count = 0;
while(!downstream.empty() && (int)downstream.peek() <A[i]){ // 하류보다 상류물고기가 더 크면 하류 물고기를 먹는다.
downstream.pop();
eat_count++;
}
if(downstream.empty()) // 상류 물고기가 하류 물고기 다 먹으면 살아서 올라감
living_count++;
}
else{ // 하류 물고기가 없을 경우
living_count++;
}
}
else{ // 물고기가 하류로 향하면
downstream.push(A[i]);
}
}
living_count += downstream.size();
return living_count;
}
}
Code: 06:23:50 UTC,
java,
verify,
result: Passed
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
// you can also use imports, for example:
import java.util.*;
// you can use System.out.println for debugging purposes, e.g.
// System.out.println("this is a debug message");
class Solution {
public int solution(int[] A, int[] B) {
// write your code in Java SE 8
/*
문제:
입력되는 배열 A의 index는 물기고번호, 값은 물고기 크기다.
입력되는 배열 B의 index는 물고기번호, 값은 물고기가 향하는 방향이다.
배열 B의 값이 0이면 상류로 가는 것,
배열 B의 값이 1이면 하류로 가는 것이다.
index의 위치가 물고기가 처음 배치된 위치이다.
그렇기 때문에 위쪽에 위치한 물고기는 아래에 있는 물고기를 만날수 없다.
또한, 같은 방향으로 향하는 물고기도 만날 수 없다.
다른 방향으로 향하는 물고기를 만나면 크기가 큰 물고기가 작은 물고리를 잡아 먹는다.
방법:
1. index가 작을수록(P < Q)일수록 상류에 위치한다.
2. 하류에 위치한 물고기에 초점을 맞추고 자신보다 아래에서 올라오는 물고기와 비교한다.
3. 살아남은 물고기 들은 스택에 저장한다.
시나리오:
0. 물고기가 하류로 향하면 스택에 넣는다.
0-1. 다음 물고기가 상류로 향하는 물고기 일 경우 스택에서 빼와 비교한다.
0-2. 상류 물고기의 크기가 크다면 비교 물고기를 먹는다(없엔다)
0-3. 상류 물고기가 작다면 스택에 그대로 있다.
1. 물고기가 상류로 향하고, 비교 물고기(하류로향하는 물고기)가 없으면 살아남은 물고기
2. 물고기가 하류로 향하고, 비교 물고기가 있으면, 비교 물고기는 하류 스택에 넣는다.
*/
int living_count = 0;
Stack downstream = new Stack<Integer>();
for(int i = 0; i<A.length; i++){
int stream = B[i];
if(B[i] == 0){ // 물고기가 상류로 향하면
// 1. 하류로 향하는 물고기가 있으면 비교, 없으면 살아서 올라감.
if(!downstream.empty()){
int eat_count = 0;
while(!downstream.empty() && (int)downstream.peek() <A[i]){ // 하류보다 상류물고기가 더 크면 하류 물고기를 먹는다.
downstream.pop();
eat_count++;
}
if(downstream.empty()) // 상류 물고기가 하류 물고기 다 먹으면 살아서 올라감
living_count++;
}
else{ // 하류 물고기가 없을 경우
living_count++;
}
}
else{ // 물고기가 하류로 향하면
downstream.push(A[i]);
}
}
living_count += downstream.size();
return living_count;
}
}
Analysis
User test case 1:
[[1, 2, 3, 4, 5], [0, 0, 0, 0, 0]]
User test case 2:
[[4, 3, 2, 1, 5], [1, 1, 1, 1, 0]]
Code: 06:24:19 UTC,
java,
verify,
result: Passed
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
// you can also use imports, for example:
import java.util.*;
// you can use System.out.println for debugging purposes, e.g.
// System.out.println("this is a debug message");
class Solution {
public int solution(int[] A, int[] B) {
// write your code in Java SE 8
/*
문제:
입력되는 배열 A의 index는 물기고번호, 값은 물고기 크기다.
입력되는 배열 B의 index는 물고기번호, 값은 물고기가 향하는 방향이다.
배열 B의 값이 0이면 상류로 가는 것,
배열 B의 값이 1이면 하류로 가는 것이다.
index의 위치가 물고기가 처음 배치된 위치이다.
그렇기 때문에 위쪽에 위치한 물고기는 아래에 있는 물고기를 만날수 없다.
또한, 같은 방향으로 향하는 물고기도 만날 수 없다.
다른 방향으로 향하는 물고기를 만나면 크기가 큰 물고기가 작은 물고리를 잡아 먹는다.
방법:
1. index가 작을수록(P < Q)일수록 상류에 위치한다.
2. 하류에 위치한 물고기에 초점을 맞추고 자신보다 아래에서 올라오는 물고기와 비교한다.
3. 살아남은 물고기 들은 스택에 저장한다.
시나리오:
0. 물고기가 하류로 향하면 스택에 넣는다.
0-1. 다음 물고기가 상류로 향하는 물고기 일 경우 스택에서 빼와 비교한다.
0-2. 상류 물고기의 크기가 크다면 비교 물고기를 먹는다(없엔다)
0-3. 상류 물고기가 작다면 스택에 그대로 있다.
1. 물고기가 상류로 향하고, 비교 물고기(하류로향하는 물고기)가 없으면 살아남은 물고기
2. 물고기가 하류로 향하고, 비교 물고기가 있으면, 비교 물고기는 하류 스택에 넣는다.
*/
int living_count = 0;
Stack downstream = new Stack<Integer>();
for(int i = 0; i<A.length; i++){
int stream = B[i];
if(B[i] == 0){ // 물고기가 상류로 향하면
// 1. 하류로 향하는 물고기가 있으면 비교, 없으면 살아서 올라감.
if(!downstream.empty()){
int eat_count = 0;
while(!downstream.empty() && (int)downstream.peek() <A[i]){ // 하류보다 상류물고기가 더 크면 하류 물고기를 먹는다.
downstream.pop();
eat_count++;
}
if(downstream.empty()) // 상류 물고기가 하류 물고기 다 먹으면 살아서 올라감
living_count++;
}
else{ // 하류 물고기가 없을 경우
living_count++;
}
}
else{ // 물고기가 하류로 향하면
downstream.push(A[i]);
}
}
living_count += downstream.size();
return living_count;
}
}
Analysis
User test case 1:
[[1, 2, 3, 4, 5], [0, 0, 0, 0, 0]]
User test case 2:
[[4, 3, 2, 1, 5], [1, 1, 1, 1, 0]]
Code: 06:24:37 UTC,
java,
verify,
result: Passed
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
// you can also use imports, for example:
import java.util.*;
// you can use System.out.println for debugging purposes, e.g.
// System.out.println("this is a debug message");
class Solution {
public int solution(int[] A, int[] B) {
// write your code in Java SE 8
/*
문제:
입력되는 배열 A의 index는 물기고번호, 값은 물고기 크기다.
입력되는 배열 B의 index는 물고기번호, 값은 물고기가 향하는 방향이다.
배열 B의 값이 0이면 상류로 가는 것,
배열 B의 값이 1이면 하류로 가는 것이다.
index의 위치가 물고기가 처음 배치된 위치이다.
그렇기 때문에 위쪽에 위치한 물고기는 아래에 있는 물고기를 만날수 없다.
또한, 같은 방향으로 향하는 물고기도 만날 수 없다.
다른 방향으로 향하는 물고기를 만나면 크기가 큰 물고기가 작은 물고리를 잡아 먹는다.
방법:
1. index가 작을수록(P < Q)일수록 상류에 위치한다.
2. 하류에 위치한 물고기에 초점을 맞추고 자신보다 아래에서 올라오는 물고기와 비교한다.
3. 살아남은 물고기 들은 스택에 저장한다.
시나리오:
0. 물고기가 하류로 향하면 스택에 넣는다.
0-1. 다음 물고기가 상류로 향하는 물고기 일 경우 스택에서 빼와 비교한다.
0-2. 상류 물고기의 크기가 크다면 비교 물고기를 먹는다(없엔다)
0-3. 상류 물고기가 작다면 스택에 그대로 있다.
1. 물고기가 상류로 향하고, 비교 물고기(하류로향하는 물고기)가 없으면 살아남은 물고기
2. 물고기가 하류로 향하고, 비교 물고기가 있으면, 비교 물고기는 하류 스택에 넣는다.
*/
int living_count = 0;
Stack downstream = new Stack<Integer>();
for(int i = 0; i<A.length; i++){
int stream = B[i];
if(B[i] == 0){ // 물고기가 상류로 향하면
// 1. 하류로 향하는 물고기가 있으면 비교, 없으면 살아서 올라감.
if(!downstream.empty()){
int eat_count = 0;
while(!downstream.empty() && (int)downstream.peek() <A[i]){ // 하류보다 상류물고기가 더 크면 하류 물고기를 먹는다.
downstream.pop();
eat_count++;
}
if(downstream.empty()) // 상류 물고기가 하류 물고기 다 먹으면 살아서 올라감
living_count++;
}
else{ // 하류 물고기가 없을 경우
living_count++;
}
}
else{ // 물고기가 하류로 향하면
downstream.push(A[i]);
}
}
living_count += downstream.size();
return living_count;
}
}
Analysis
User test case 1:
[[1, 2, 3, 4, 5], [0, 0, 0, 0, 0]]
User test case 2:
[[4, 3, 2, 1, 5], [0, 1, 0, 0, 0]]
Code: 06:24:51 UTC,
java,
verify,
result: Passed
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
// you can also use imports, for example:
import java.util.*;
// you can use System.out.println for debugging purposes, e.g.
// System.out.println("this is a debug message");
class Solution {
public int solution(int[] A, int[] B) {
// write your code in Java SE 8
/*
문제:
입력되는 배열 A의 index는 물기고번호, 값은 물고기 크기다.
입력되는 배열 B의 index는 물고기번호, 값은 물고기가 향하는 방향이다.
배열 B의 값이 0이면 상류로 가는 것,
배열 B의 값이 1이면 하류로 가는 것이다.
index의 위치가 물고기가 처음 배치된 위치이다.
그렇기 때문에 위쪽에 위치한 물고기는 아래에 있는 물고기를 만날수 없다.
또한, 같은 방향으로 향하는 물고기도 만날 수 없다.
다른 방향으로 향하는 물고기를 만나면 크기가 큰 물고기가 작은 물고리를 잡아 먹는다.
방법:
1. index가 작을수록(P < Q)일수록 상류에 위치한다.
2. 하류에 위치한 물고기에 초점을 맞추고 자신보다 아래에서 올라오는 물고기와 비교한다.
3. 살아남은 물고기 들은 스택에 저장한다.
시나리오:
0. 물고기가 하류로 향하면 스택에 넣는다.
0-1. 다음 물고기가 상류로 향하는 물고기 일 경우 스택에서 빼와 비교한다.
0-2. 상류 물고기의 크기가 크다면 비교 물고기를 먹는다(없엔다)
0-3. 상류 물고기가 작다면 스택에 그대로 있다.
1. 물고기가 상류로 향하고, 비교 물고기(하류로향하는 물고기)가 없으면 살아남은 물고기
2. 물고기가 하류로 향하고, 비교 물고기가 있으면, 비교 물고기는 하류 스택에 넣는다.
*/
int living_count = 0;
Stack downstream = new Stack<Integer>();
for(int i = 0; i<A.length; i++){
int stream = B[i];
if(B[i] == 0){ // 물고기가 상류로 향하면
// 1. 하류로 향하는 물고기가 있으면 비교, 없으면 살아서 올라감.
if(!downstream.empty()){
int eat_count = 0;
while(!downstream.empty() && (int)downstream.peek() <A[i]){ // 하류보다 상류물고기가 더 크면 하류 물고기를 먹는다.
downstream.pop();
eat_count++;
}
if(downstream.empty()) // 상류 물고기가 하류 물고기 다 먹으면 살아서 올라감
living_count++;
}
else{ // 하류 물고기가 없을 경우
living_count++;
}
}
else{ // 물고기가 하류로 향하면
downstream.push(A[i]);
}
}
living_count += downstream.size();
return living_count;
}
}
Analysis
User test case 1:
[[1, 2, 3, 4, 5], [0, 1, 0, 0, 0]]
User test case 2:
[[4, 3, 2, 1, 5], [0, 1, 0, 0, 0]]
Code: 06:25:07 UTC,
java,
verify,
result: Passed
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
// you can also use imports, for example:
import java.util.*;
// you can use System.out.println for debugging purposes, e.g.
// System.out.println("this is a debug message");
class Solution {
public int solution(int[] A, int[] B) {
// write your code in Java SE 8
/*
문제:
입력되는 배열 A의 index는 물기고번호, 값은 물고기 크기다.
입력되는 배열 B의 index는 물고기번호, 값은 물고기가 향하는 방향이다.
배열 B의 값이 0이면 상류로 가는 것,
배열 B의 값이 1이면 하류로 가는 것이다.
index의 위치가 물고기가 처음 배치된 위치이다.
그렇기 때문에 위쪽에 위치한 물고기는 아래에 있는 물고기를 만날수 없다.
또한, 같은 방향으로 향하는 물고기도 만날 수 없다.
다른 방향으로 향하는 물고기를 만나면 크기가 큰 물고기가 작은 물고리를 잡아 먹는다.
방법:
1. index가 작을수록(P < Q)일수록 상류에 위치한다.
2. 하류에 위치한 물고기에 초점을 맞추고 자신보다 아래에서 올라오는 물고기와 비교한다.
3. 살아남은 물고기 들은 스택에 저장한다.
시나리오:
0. 물고기가 하류로 향하면 스택에 넣는다.
0-1. 다음 물고기가 상류로 향하는 물고기 일 경우 스택에서 빼와 비교한다.
0-2. 상류 물고기의 크기가 크다면 비교 물고기를 먹는다(없엔다)
0-3. 상류 물고기가 작다면 스택에 그대로 있다.
1. 물고기가 상류로 향하고, 비교 물고기(하류로향하는 물고기)가 없으면 살아남은 물고기
2. 물고기가 하류로 향하고, 비교 물고기가 있으면, 비교 물고기는 하류 스택에 넣는다.
*/
int living_count = 0;
Stack downstream = new Stack<Integer>();
for(int i = 0; i<A.length; i++){
int stream = B[i];
if(B[i] == 0){ // 물고기가 상류로 향하면
// 1. 하류로 향하는 물고기가 있으면 비교, 없으면 살아서 올라감.
if(!downstream.empty()){
int eat_count = 0;
while(!downstream.empty() && (int)downstream.peek() <A[i]){ // 하류보다 상류물고기가 더 크면 하류 물고기를 먹는다.
downstream.pop();
eat_count++;
}
if(downstream.empty()) // 상류 물고기가 하류 물고기 다 먹으면 살아서 올라감
living_count++;
}
else{ // 하류 물고기가 없을 경우
living_count++;
}
}
else{ // 물고기가 하류로 향하면
downstream.push(A[i]);
}
}
living_count += downstream.size();
return living_count;
}
}
Analysis
Code: 06:25:20 UTC,
java,
final,
score:
100
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
// you can also use imports, for example:
import java.util.*;
// you can use System.out.println for debugging purposes, e.g.
// System.out.println("this is a debug message");
class Solution {
public int solution(int[] A, int[] B) {
// write your code in Java SE 8
/*
문제:
입력되는 배열 A의 index는 물기고번호, 값은 물고기 크기다.
입력되는 배열 B의 index는 물고기번호, 값은 물고기가 향하는 방향이다.
배열 B의 값이 0이면 상류로 가는 것,
배열 B의 값이 1이면 하류로 가는 것이다.
index의 위치가 물고기가 처음 배치된 위치이다.
그렇기 때문에 위쪽에 위치한 물고기는 아래에 있는 물고기를 만날수 없다.
또한, 같은 방향으로 향하는 물고기도 만날 수 없다.
다른 방향으로 향하는 물고기를 만나면 크기가 큰 물고기가 작은 물고리를 잡아 먹는다.
방법:
1. index가 작을수록(P < Q)일수록 상류에 위치한다.
2. 하류에 위치한 물고기에 초점을 맞추고 자신보다 아래에서 올라오는 물고기와 비교한다.
3. 살아남은 물고기 들은 스택에 저장한다.
시나리오:
0. 물고기가 하류로 향하면 스택에 넣는다.
0-1. 다음 물고기가 상류로 향하는 물고기 일 경우 스택에서 빼와 비교한다.
0-2. 상류 물고기의 크기가 크다면 비교 물고기를 먹는다(없엔다)
0-3. 상류 물고기가 작다면 스택에 그대로 있다.
1. 물고기가 상류로 향하고, 비교 물고기(하류로향하는 물고기)가 없으면 살아남은 물고기
2. 물고기가 하류로 향하고, 비교 물고기가 있으면, 비교 물고기는 하류 스택에 넣는다.
*/
int living_count = 0;
Stack downstream = new Stack<Integer>();
for(int i = 0; i<A.length; i++){
int stream = B[i];
if(B[i] == 0){ // 물고기가 상류로 향하면
// 1. 하류로 향하는 물고기가 있으면 비교, 없으면 살아서 올라감.
if(!downstream.empty()){
int eat_count = 0;
while(!downstream.empty() && (int)downstream.peek() <A[i]){ // 하류보다 상류물고기가 더 크면 하류 물고기를 먹는다.
downstream.pop();
eat_count++;
}
if(downstream.empty()) // 상류 물고기가 하류 물고기 다 먹으면 살아서 올라감
living_count++;
}
else{ // 하류 물고기가 없을 경우
living_count++;
}
}
else{ // 물고기가 하류로 향하면
downstream.push(A[i]);
}
}
living_count += downstream.size();
return living_count;
}
}