coding_test_study/210221_BOJ_12891.cpp at main · flylofty/coding_test_study · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#include <iostream>
#include <string>
#include <vector>

using namespace std;

int S, P;
vector<int> minimum(4 ,0);
vector<int> cnt(4, 0);

bool isGood() {
	for (int i = 0; i < 4; ++i) {
		if (cnt[i] < minimum[i])
			return false;
	}
	return true;
}

int indexNumber(char ch) {
	if (ch == 'A')
		return 0;
	else if (ch == 'C')
		return 1;
	else if (ch == 'G')
		return 2;
	else if (ch == 'T')
		return 3;
}

int main() {

	cin >> S >> P;

	string str; cin >> str;

	for (int i = 0; i < 4; ++i) {
		cin >> minimum[i];
	}

	for (int i = 0; i < P; ++i) {
		int insert = indexNumber(str[i]);
		cnt[insert]++;
	}

	int result = 0;

	if (isGood())
		result++;

	for (int i = 1; i <= str.size() - P; ++i) {
		int remove = indexNumber(str[i - 1]);
		cnt[remove]--;
		int insert = indexNumber(str[i - 1 + P]);
		cnt[insert]++;
		if (isGood())
			result++;
	}

	cout << result << "\n";

	return 0;
}

// 연속된 부분문자열이라고 생각하지 않고 작성한 코드;;;;;
//#include <iostream>
//#include <string>
//#include <vector>
//
//using namespace std;
//
//int S, P;
//vector<char> arr{ 'A', 'C', 'G', 'T' };
//vector<int> store(4, 0);
//vector<int> minimum;
//
//int main() {
//
//	cin >> S >> P;
//
//	string str; cin >> str;
//
//	for (int i = 0; i < str.size(); ++i) {
//		if (str[i] == 'A')
//			store[0]++;
//		else if (str[i] == 'C')
//			store[1]++;
//		else if (str[i] == 'G')
//			store[2]++;
//		else if (str[i] == 'T')
//			store[3]++;
//	}
//
//	for (int i = 0; i < 4; ++i) {
//		int num; cin >> num;
//
//		if (store[i] < num) {
//			cout << "0\n";
//			return 0;
//		}
//	}
//
//	return 0;
//}