codeforces #597 div2 C - Constanze's Machine ( 피보나치 , 구현)

https://codeforces.com/contest/1245/problem/C

 

단어를 입력하면 m은 nn으로, w는 uu로 바꿔서 출력하는 기계가 있다.

(출력에 m이나, w가 나올 수 없다. 이경우 0 출력)

 

예를들어 단어에 nnn이라는 문자배열이 있다면, 가능한 입력은 nnn mn nm 총 세가지가 존재한다.

 

'n'이 k번 연속한 문자배열을 만들 수 있는 입력은 피보나치 수열을 이루게 된다.

(왜 그런지는 모르겠다. 몇번 해보니까 규칙이 보임)

 

따라서 단어에서 n이 나오는 idx를 기록한 idx_n과 u가 나오는 idx를 기록한 idx_u에 대해서 적절히 연산해 주었다.

 

#include <iostream>
#include <algorithm>
#include <string>
using namespace std;
const int MAX = 1e5+1;
const int MOD = 1e9 + 7;
typedef long long ll;

int main() {
	string str; cin >> str;
	int idx_u[MAX], usize=0, idx_n[MAX], nsize=0;
	for (int i = 0; i < str.size(); i++) {
		if (str[i] == 'u') 
			idx_u[usize++] = i;
		if (str[i] == 'n')
			idx_n[nsize++] = i;		
	}

	ll d[MAX] = { 0 };
	d[1] = 1, d[2] = 2;
	for (int i = 3; i <= MAX-1; i++)
		d[i] = (d[i - 1] + d[i - 2]) % MOD;


	ll t = 1, ans_u = 1, ans_n=1;
	for (int i = 1; i <= usize; i++) {
		if (i<usize && idx_u[i] - idx_u[i - 1] == 1)
			t++;
		else {
			ans_u = (ans_u * d[t])%MOD;
			t = 1;
		}
	}
	for (int i = 1; i <= nsize; i++) {
		if (i<nsize && idx_n[i] - idx_n[i - 1] == 1)
			t++;
		else {
			ans_n = (ans_n * d[t]) % MOD;
			t = 1;
		}
	}
	if (str.find("m") != string::npos || str.find("w") != string::npos)
		cout << 0 << '\n';
	else
		cout << (ans_u * ans_n) % MOD << '\n';
}

 

아래는 editorial. 굉장히 간단하다.

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    
    string s;
    cin >> s;
    
    for (char x : s)
    {
        if (x == 'w' || x == 'm')
        {
            cout << 0 << '\n';
            return 0;
        }
    }
    
    int n = s.size();
    vector<int> dp(n + 1);
    dp[0] = 1;
    dp[1] = 1;
    for (int i = 2; i <= n; ++i)
    {
        dp[i] = dp[i - 1];
        if (s[i - 1] == s[i - 2] && (s[i - 1] == 'u' || s[i - 1] == 'n'))
            dp[i] = (dp[i] + dp[i - 2]) % MOD;
    }
    
    cout << dp[n] << '\n';
    
    return 0;
}