Thứ Năm, 3 tháng 1, 2019

Codeforces Round 57- Bài 2


Bài 2:  You are given a string $s$ of length $n$ consisting only of lowercase Latin letters
A substring of a string is a contiguous subsequence of that string. So, string "forces" is substring of string "codeforces", but string "coder" is not.
Your task is to calculate the number of ways to remove exactly one substring from this string in such a way that all remaining characters are equal (the number of distinct characters either zero or one).
It is guaranteed that there is at least two different characters in s.
Note that you can remove the whole string and it is correct. Also note that you should remove at least one character.
Since the answer can be rather large (not very large though) print it modulo 998244353.
If you are Python programmer, consider using PyPy instead of Python when you submit your code.
Input
The first line of the input contains one integer $n(2\le n\le 2.10^5)$ - the length of the string s. 
The second line of the input contains the string $s$ of length $n$ consisting only of lowercase Latin letters
It is guaranteed that there is at least two different characters in $s$.
Output
Print one integer — the number of ways modulo 998244353 to remove exactly one substring from $s$ in such way that all remaining characters are equal.
Examples
input
Copy
4
abaa
output
Copy
6
input
Copy
7
aacdeee
output
Copy
6
Solution:
#include <bits/stdc++.h>
using namespace std;
int64_t n,i,j;
string s;
int main(){
 cin>>n>>s;
 for(;i<n&&s[i]==s[0];i++);
 for(;j<n&&s[n-1-j]==s[n-1];j++);
 cout<<(s[0]!=s[n-1]?i+j+1:i+j<n?(i+1)*(j+1):n*(n+1)/2)%998244353;
}

Không có nhận xét nào:

Đăng nhận xét

Bài G - Educatioal Round 62

Đề bài: Bạn được cho 1 đồ thị vô hướng đặc biệt. Nó bao gồm $2n$ đỉnh được đánh số từ 1 đến 2n. Dưới đây là một số đặc tính của đồ thị: + ...