Osajono

import java.util.ArrayList;

public class teht1 {
	public static void main(String[] args) {
		IO io = new IO();
		
		String string = io.next();
		ArrayList<String> osajonot = new ArrayList<String>();

		String[] string2 = string.split("");
		int j = string.length();
		
		for(int i=1; i<=string.length(); i++) {
			for(int k=0; k<j; k++) {
				String s = "";
				for(int n=0; n<i; n++) {
					s += string2[k + n];
				}
				osajonot.add(s);
			}
			j--;
		}
		
		int result = 0;
		
		for(String osajono : osajonot) {
			String[] split = osajono.split("");
			if(split[0].equals(split[0 + split.length - 1])) {
				result++;
			}
		}
		io.print(result);
		
		io.close();
	}
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED