-
Notifications
You must be signed in to change notification settings - Fork 7
/
Main.java
31 lines (28 loc) · 896 Bytes
/
Main.java
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
package com.company;
public class Main {
public static void main(String[] args) {
String s = "aba";
System.out.println(countSubstrings(s));
}
public static int countSubstrings(String s) {
int count = s.length();
for (int i = 0; i < s.length(); i++) {
int li = i - 1, hi = i + 1;
while (li >= 0 && hi < s.length()) {
if (s.charAt(li--) == s.charAt(hi++)) count++;
else break;
}
}
for (int i = 0; i < s.length() - 1; i++) {
if (s.charAt(i) == s.charAt(i + 1)) {
count++;
int li = i - 1, hi = i + 2;
while (li >= 0 && hi < s.length()) {
if (s.charAt(li--) == s.charAt(hi++)) count++;
else break;
}
}
}
return count;
}
}