【HDOJ】1332 LC-Display

水题。

#include <cstdio>
#include <cstring>
#include <cstdlib>

#define MAXN 11
#define LEFT  1
#define RIGHT 2
#define BOTH  3

char lcd[10][MAXN][2*MAXN+3+1][MAXN+2+1];
int rr[MAXN];
int cc[MAXN];

void addR(int x, int i, int j) {
    int k;

    for (k=1; k<cc[i]-1; ++k)
        lcd[x][i][j][k] = '-';
}

void addC(int x, int i,int j, int v) {
    if (v & LEFT)
        lcd[x][i][j][0] = '|';
    if (v & RIGHT)
        lcd[x][i][j][cc[i]-1] = '|';
}

void init() {
    int i, j, k;
    int n;

    for (i=1; i<=10; ++i) {
        rr[i] = 2*i+3;
        cc[i] = i+2;
    }

    memset(lcd, ' ', sizeof(lcd));

    for (k=0; k<10; ++k) {
        for (i=1; i<=10; ++i) {
            for (j=0; j<rr[i]; ++j) {
                lcd[k][i][j][cc[i]] = '';
            }
        }
    }

    // 0
    n = 0;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0 || j==rr[i]-1) {
                addR(n, i, j);
            } else if (j == i+1) {
                /*do nothing*/;
            } else {
                addC(n, i, j, BOTH);
            }
        }
    }

    // 1
    n = 1;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0 || j==i+1 || j==rr[i]-1) {
                /*do nothing*/;
            } else {
                addC(n, i, j, RIGHT);
            }
        }
    }

    // 2
    n = 2;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0 || j==i+1 || j==rr[i]-1) {
                addR(n, i, j);
            } else {
                if (j < i+1)
                    addC(n, i, j, RIGHT);
                else
                    addC(n, i, j, LEFT);
            }
        }
    }

    // 3
    n = 3;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0 || j==i+1 || j==rr[i]-1) {
                addR(n, i, j);
            } else {
                addC(n, i, j, RIGHT);
            }
        }
    }

    // 4
    n = 4;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0 || j==rr[i]-1) {
                /*do nothing*/;
            } else if (j == i+1) {
                addR(n, i, j);
            } else {
                if (j < i+1)
                    addC(n, i, j, BOTH);
                else
                    addC(n, i, j, RIGHT);
            }
        }
    }

    // 5
    n = 5;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0 || j==i+1 || j==rr[i]-1) {
                addR(n, i, j);
            } else {
                if (j < i+1)
                    addC(n, i, j, LEFT);
                else
                    addC(n, i, j, RIGHT);
            }
        }
    }

    // 6
    n = 6;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0 || j==i+1 || j==rr[i]-1) {
                addR(n, i, j);
            } else {
                if (j < i+1)
                    addC(n, i, j, LEFT);
                else
                    addC(n, i, j, BOTH);
            }
        }
    }

    // 7
    n = 7;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0) {
                addR(n, i, j);
            } else if (j==i+1 || j==rr[i]-1) {
                /*do nothing*/;
            } else {
                addC(n, i, j, RIGHT);
            }
        }
    }

    // 8
    n = 8;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0 || j==i+1 || j==rr[i]-1) {
                addR(n, i, j);
            } else {
                addC(n, i, j, BOTH);
            }
        }
    }

    // 9
    n = 9;
    for (i=1; i<=10; ++i) {
        for (j=0; j<rr[i]; ++j) {
            if (j==0 || j==i+1 || j==rr[i]-1) {
                addR(n, i, j);
            } else {
                if (j < i+1)
                    addC(n, i, j, BOTH);
                else
                    addC(n, i, j, RIGHT);
            }
        }
    }
}

void output(int n) {
    int i, j, k;

    for (i=0; i<10; ++i) {
        for (j=0; j<rr[n]; ++j) {
            printf("%s
", lcd[i][n][j]);
        }
        printf("

");
    }
    printf("output finish
");
}

int main() {
    int n;
    int i, j, k;
    char s[15];

#ifndef ONLINE_JUDGE
    freopen("data.in", "r", stdin);
    freopen("data.out", "w", stdout);
#endif

    init();
    //output(1);

    while (scanf("%d %s", &n, s) != EOF) {
        if (n==0 && s[0]=='0' && s[1]=='')
            break;
        for (j=0; j<rr[n]; ++j) {
            for (i=0; s[i]; ++i) {
                if (i)
                    printf(" %s", lcd[s[i]-'0'][n][j]);
                else
                    printf("%s", lcd[s[i]-'0'][n][j]);
            }
            printf("
");
        }
        printf("
");
    }

    return 0;
}