www.acmicpc.net/problem/5430
5430번: AC
각 테스트 케이스에 대해서, 입력으로 주어진 정수 배열에 함수를 수행한 결과를 출력한다. 만약, 에러가 발생한 경우에는 error를 출력한다.
www.acmicpc.net
문제
선영이는 주말에 할 일이 없어서 새로운 언어 AC를 만들었다. AC는 정수 배열에 연산을 하기 위해 만든 언어이다. 이 언어에는 두 가지 함수 R(뒤집기)과 D(버리기)가 있다.
함수 R은 배열에 있는 숫자의 순서를 뒤집는 함수이고, D는 첫 번째 숫자를 버리는 함수이다. 배열이 비어있는데 D를 사용한 경우에는 에러가 발생한다.
함수는 조합해서 한 번에 사용할 수 있다. 예를 들어, "AB"는 A를 수행한 다음에 바로 이어서 B를 수행하는 함수이다. 예를 들어, "RDD"는 배열을 뒤집은 다음 처음 두 숫자를 버리는 함수이다.
배열의 초기값과 수행할 함수가 주어졌을 때, 최종 결과를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 테스트 케이스의 개수 T가 주어진다. T는 최대 100이다.
각 테스트 케이스의 첫째 줄에는 수행할 함수 p가 주어진다. p의 길이는 1보다 크거나 같고, 100,000보다 작거나 같다.
다음 줄에는 배열에 들어있는 수의 개수 n이 주어진다. (0 ≤ n ≤ 100,000)
다음 줄에는 [x1,...,xn]과 같은 형태로 배열에 들어있는 수가 주어진다. (1 ≤ xi ≤ 100)
전체 테스트 케이스에 주어지는 p의 길이의 합과 n의 합은 70만을 넘지 않는다.
출력
각 테스트 케이스에 대해서, 입력으로 주어진 정수 배열에 함수를 수행한 결과를 출력한다. 만약, 에러가 발생한 경우에는 error를 출력한다.
아래 코드 완성해서 정상적으로 출력되는 데 런타임오류(segfault)가 발생한다.
어디가 문제인지 잘 모르겠네요.
#include <iostream>
#include <deque>
#include <cstring>
using namespace std;
deque<int > q;
#define MAX 100001
int main ()
{
ios_base::sync_with_stdio (0 );
cin.tie (0 );
cout.tie (0 );
int i=0 ,k=0 ,m;
int testN;
int N;
char dt[MAX];
char control[MAX];
int numC, sizeq;
int flag;
int error=0 ;
cin >> testN;
for (i =0 ; i < testN; i++) {
error = 0 ;
cin >> control;
cin >> N;
cin >> dt;
char *tok=strtok (dt, "[,]" );
flag = 1 ;
while (tok != NULL ) {
q.push_back (atoi (tok));
tok=strtok (NULL , "[,]" );
}
numC = strlen (control);
for (k=0 ; k< numC; k++ ) {
if ( control[k] == 'R' ) {
flag = (flag + 1 )%2 ;
} else if ( control[k] == 'D' ){
if (!q.empty ()){
if ( flag == true ){
q.pop_front ();
}else if ( flag == false ){
q.pop_back ();
}
} else {
error++ ;
continue ;
}
}
}
if (!q.empty () && error == 0 ) {
cout << '[' ;
sizeq = q.size ();
for (m=0 ; m<sizeq; m++){
if ( !q.empty () && (flag == true ) ){
if ( q.size () > 1 ){
cout << q.front () << ',' ;
}else if (q.size () == 1 ) {
cout << q.front ();
}
q.pop_front ();
}else if ( !q.empty () && (flag == false ) ) {
if ( q.size () > 1 ){
cout << q.back () << ',' ;
}else if ( q.size () == 1 ) {
cout << q.back ();
}
q.pop_back ();
}
}
cout << ']' << '\n' ;
} else if (error >= 1 ){
cout << "error" << '\n' ;
}else if (q.size ()==0 ){
cout << "[]\n" ;
}
}
return 0 ;
}
문제 해결이 됐습니다. control은 RD와 같이 control 입력부이고 dt는 숫자 array이 인데 cotrol보다 3배 이상 크게 잡아 줘야 문제 해결이 됩니다.
char control[MAX]; //RD control 입력 int N; // dt array size char dt[MAX*3+2];// array [1,2,3,4]
수정코드는 아래와 같습니다. 참고하시기 바랍니다.
#include <iostream>
#include <deque>
#include <cstring>
using namespace std;
deque<int > q;
#define MAX 100001
int main ()
{
ios_base::sync_with_stdio (0 );
cin.tie (0 );
cout.tie (0 );
int i=0 ,k=0 ,m;
int testN;
char control[MAX];
int N;
char dt[MAX*3 +2 ];
int numC, sizeq;
int flag;
int error=0 ;
cin >> testN;
for (i =0 ; i < testN; i++) {
error = 0 ;
cin >> control;
cin >> N;
cin >> dt;
char *tok=strtok (dt, "[,]" );
flag = 1 ;
while (tok != NULL ) {
q.push_back (atoi (tok));
tok=strtok (NULL , "[,]" );
}
numC = strlen (control);
for (k=0 ; k< numC; k++ ) {
if ( control[k] == 'R' ) {
flag = (flag + 1 )%2 ;
} else if ( control[k] == 'D' ){
if (!q.empty ()){
if ( flag == true ){
q.pop_front ();
}else if ( flag == false ){
q.pop_back ();
}
} else {
error++ ;
continue ;
}
}
}
if (!q.empty () && error == 0 ) {
cout << '[' ;
sizeq = q.size ();
for (m=0 ; m<sizeq; m++){
if ( !q.empty () && (flag == true ) ){
if ( q.size () > 1 ){
cout << q.front () << ',' ;
}else if (q.size () == 1 ) {
cout << q.front ();
}
q.pop_front ();
}else if ( !q.empty () && (flag == false ) ) {
if ( q.size () > 1 ){
cout << q.back () << ',' ;
}else if ( q.size () == 1 ) {
cout << q.back ();
}
q.pop_back ();
}
}
cout << ']' << '\n' ;
} else if (error >= 1 ){
cout << "error" << '\n' ;
}else if (q.size ()==0 ){
cout << "[]\n" ;
}
}
return 0 ;
}