yukicoder No.22 - 括弧の対応

Source

ニコニコミュニティ
問題文

問題概要

$\verb|(|$ と $\verb|)|$ のみからなる $N$ 文字の文字列 $S$ が与えられる.
ここで,$S$ において,カッコの対応はきちんと取れていると仮定して良い.
$S$ において,$K$ 文字目と対応するのは何文字目かを求める問題.

解法

$K$ 文字目が $\verb|(|$ とする($\verb|)|$ の場合もほとんど同様).
最初カッコの深さを $0$ にセットして,$K$ 文字目から初めて,$\verb|(|$ が現れるたびに深さを $+1$ して,$\verb|)|$ が現れるたびに深さを $-1$ して,はじめに深さが $0$ になった場所が答え.

C++によるスパゲッティなソースコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void reader(int *x, int *y){reader(x);reader(y);}
int reader(char c[]){int i,s=0;for(;;){mygc(i);if(i!=' '&&i!='\n'&&i!='\r'&&i!='\t'&&i!=EOF) break;}c[s++]=i;for(;;){mygc(i);if(i==' '||i=='\n'||i=='\r'||i=='\t'||i==EOF) break;c[s++]=i;}c[s]='\0';return s;}
void writer(int x, char c){int s=0,m=0;char f[10];if(x<0)m=1,x=-x;while(x)f[s++]=x%10,x/=10;if(!s)f[s++]=0;if(m)mypc('-');while(s--)mypc(f[s]+'0');mypc(c);}

int N, K; char S[11000];

int main(){
  int i, j, k;

  reader(&N,&K); K--;
  reader(S);

  if(S[K]=='('){
    k = 0;
    for(i=K;;i++){
      if(S[i]=='(') k++; else k--;
      if(k==0) break;
    }
  } else {
    k = 0;
    for(i=K;;i--){
      if(S[i]=='(') k++; else k--;
      if(k==0) break;
    }
  }

  writer(i+1,'\n');

  return 0;
}

Current time: 2017年09月25日07時54分39秒
Last modified: 2014年11月17日00時42分41秒 (by laycrs)
Tags: Competitive_Programming yukicoder
トップページに戻る

Logged in as: unknown user (not login)

ログイン: