Osajono

import java.util.ArrayList;
import java.util.Arrays;

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

		int j = string.length();
		int result = 0;
		
		for(int i=1; i<=string.length(); i++) {
			for(int k=0; k<j; k++) {
				StringBuilder s = new StringBuilder();
				for(int n=0; n<i; n++) {
					s.append(string.substring((k + n),k + n + 1));
				}
				if(s.substring(0,1).equals(s.substring(0 + s.length() - 1, 0 + s.length()))) {
					result++;
				}
			}
			j--;
		}
		

		io.print(result + "\n");
		
		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