AtCoder Beginner Contest #031 D問題 - 語呂合わせ

Source

AtCoder Beginner Contest #031
問題文

問題概要

$1$ から $K$ までの数字が $1$ 文字以上 $3$ 文字以下のアルファベット小文字からなる文字列と対応している関係がある.
数字の列は,対応するアルファベットの文字列を連結したものに対応する.
$N$ 個の数字の列 $V_i$ と,アルファベットの文字列 $W_i$ との対応関係($i=1,2,\ldots,N$)が与えられるので,各数字がどのような文字列に対応するかを求める問題.
答えが存在することは保証されており,複数ある場合はどれを出力しても良い.

解法

各数字が何文字のアルファベットに対応するかを全探索して,整合性が取れているかをチェックする.
時間計算量は $O(3^K (K+\sum V_i))$ 程度.

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);}
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;}
template <class T, class S> void reader(T *x, S *y){reader(x);reader(y);}
void writer(const char x[], char c){int i;for(i=0;x[i]!='\0';i++)mypc(x[i]);mypc(c);}
template<class T> void writerLn(T x){writer(x,'\n');}

int K, N;
char V[50][100], W[50][100];
int Vs[50], Ws[50];

int pw[20];
int len[20];
string res[20];

int main(){
  int i, j, k, x, d, id, mask;
  string tmp;

  reader(&K,&N);
  rep(i,N){
    Vs[i] = reader(V[i]);
    Ws[i] = reader(W[i]);
    rep(j,Vs[i]) V[i][j] -= '1';
  }

  pw[0] = 1;
  REP(i,1,20) pw[i] = pw[i-1] * 3;

  rep(mask,pw[K]){
    k = mask;
    rep(i,K){
      len[i] = 1 + k%3;
      k /= 3;
    }

    rep(i,N){
      x = 0;
      rep(j,Vs[i]) x += len[V[i][j]];
      if(x != Ws[i]) break;
    }
    if(i<N) continue;

    d = 0;
    rep(i,K) res[i] = "";
    rep(i,K) rep(j,len[i]) res[i] += ' ';
    rep(i,N){
      x = 0;
      rep(j,Vs[i]){
        id = V[i][j];
        rep(k,len[id]){
          if(res[id][k]==' ') res[id][k] = W[i][x];
          if(res[id][k] != W[i][x]) d = 1;
          x++;
        }
      }
      if(d) break;
    }
    if(d) continue;

    rep(i,K) rep(j,len[i]) if(res[i][j]==' ') res[i][j] = 'a';

    rep(i,K) writerLn(res[i].c_str());
    break;
  }

  return 0;
}

Current time: 2017年07月25日05時34分57秒
Last modified: 2015年11月22日07時54分03秒 (by laycrs)
Tags: Competitive_Programming AtCoder AtCoder_Beginner_Contest ABC031 ABC_D
トップページに戻る

Logged in as: unknown user (not login)

ログイン: