-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCandies
46 lines (45 loc) · 1.35 KB
/
Candies
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
//https://www.hackerrank.com/challenges/candies/problem
import java.io.*;
import java.math.*;
import java.security.*;
import java.text.*;
import java.util.*;
import java.util.concurrent.*;
import java.util.regex.*;
public class Solution
{
public static void main(String[] args)
{
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int a[] = new int[n];
int b[] = new int[n];
long sum=0;
for(int i=0;i<n;i++)
{
a[i] = sc.nextInt();
b[i] = 1;
}
//left to right check
for(int i=0;i<n-1;i++)
{
if(a[i]<a[i+1] && b[i]>=b[i+1])
b[i+1] = b[i]+1;
else if(a[i]>a[i+1] && b[i]<=b[i+1])
b[i]=b[i+1]+1;
}
//right to left check
for(int i=n-1;i>=1;i--)
{
if(a[i]<a[i-1] && b[i]>=b[i-1])
b[i-1] = b[i]+1;
else if(a[i]>a[i-1] && b[i]<=b[i-1])
b[i]=b[i-1]+1;
}
for(int i=0;i<n;i++)
{
sum+=b[i];
}
System.out.println(sum);
}
}