目录
牛客_mari和shiny_线性dp
题目解析
C++代码
Java代码
牛客_mari和shiny_线性dp
mari和shiny (nowcoder.com)
描述:
mari每天都非常shiny。她的目标是把正能量传达到世界的每个角落!
有一天,她得到了一个仅由小写字母组成的字符串。
她想知道,这个字符串有多少个"shy"的子序列?
(所谓子序列的含义见样例说明)
题目解析
简单线性 dp: 维护 i 位置之前,⼀共有多少个 "s" "sh" ,然后更新 "shy" 的个数。
C++代码
#include <iostream>
#include <string>
using namespace std;
int main()
{
int n = 0;
string str;
cin >> n >> str;
long long s = 0, h = 0, y = 0;
for(int i = 0; i < n; i++)
{
char ch = str[i];
if(ch == 's')
{
s++;
}
else if(ch == 'h')
{
h += s;
}
else if(ch == 'y')
{
y += h;
}
}
cout << y << endl;
return 0;
}
Java代码
import java.util.*;
public class Main
{
public static void main(String[] args)
{
Scanner in = new Scanner(System.in);
int n = in.nextInt();
char[] str = in.next().toCharArray();
long s = 0, h = 0, y = 0;
for(int i = 0; i < n; i++)
{
char ch = str[i];
if(ch == 's')
{
s += 1;
}
else if(ch == 'h')
{
h += s;
}
else if(ch == 'y')
{
y += h;
}
}
System.out.println(y);
}
}