ZOJ - 3954 Seven-Segment Display 【状态标记】

题目链接

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3954

题意

有一块 LED 灯

然后上面有七块灯管
在显示不同数字的时候 由灯管的开关组合 来表现一个数字

给出一个标准表
然后 输入 一个表
求这个表 的 列 是可以交换的

然后 判断能否有一种经过若干次交换后的情况
这个表能够跟 标准表 相同
如果能 就输出 YES
不能 就输出 NO

思路

我们可以用MAP 存下 标准表的列 和 输入的表 的列

然后判断一下 两个MAP 是否完全相同

然后 输入的时候 要用 scanf 用 int 保存 不然会 超时

AC代码

#include <cstdio>
#include <cstring>
#include <ctype.h>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <map>
#include <stack>
#include <set>
#include <numeric>
#include <sstream>
#include <iomanip>
#include <limits>

#define CLR(a) memset(a, 0, sizeof(a))
#define pb push_back

using namespace std;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair <int, int> pii;
typedef pair <ll, ll> pll;
typedef pair<string, int> psi;
typedef pair<string, string> pss;

const double PI = 3.14159265358979323846264338327;
const double E = exp(1);
const double eps = 1e-30;

const int INF = 0x3f3f3f3f;
const int maxn = 1e4 + 5;
const int MOD = 1e9 + 7;

int G[10][10];
int v[10][10];

struct Node
{
    int n;
    int v[10];
}q[10];

bool comp(Node x, Node y)
{
    return x.n < y.n;
}

void init()
{
    CLR(G);
    G[1][1] = G[1][4] = G[1][5] = G[1][6] = G[1][7] = 1;
    G[2][3] = G[2][6] = 1;
    G[3][5] = G[3][6] = 1;
    G[4][1] = G[4][4] = G[4][5] = 1;
    G[5][2] = G[5][5] = 1;
    G[6][2] = 1;
    G[7][4] = G[7][5] = G[7][6] = G[7][7] = 1;
    G[9][5] = 1;
}

int main()
{
    init();
    int t;
    scanf("%d", &t);
    while (t--)
    {
        CLR(v);
        int n;
        int arr[9];
        scanf("%d", &n);
        for (int i = 0; i < n; i++)
        {
            scanf("%d", &q[i].n);
            for (int j = 0; j < 7; j++)
                scanf("%1d", &q[i].v[j]);
        }
        sort(q, q + n, comp);
        map <int, int> vis[2];
        int num = 0;
        for (int i = 0; i < 7; i++)
        {
            num = 0;
            for (int j = 0; j < n; j++)
            {
                num = num * 2 + q[j].v[i];
            }
            vis[0][num]++;
        }
        for (int i = 1; i <= 7; i++)
        {
            num = 0;
            for (int j = 0; j < n; j++)
            {
                num = num * 2 + G[q[j].n][i];
            }
            vis[1][num]++;
        }
        map <int, int>::iterator it;
        int flag = 1;
        for (it = vis[0].begin(); it != vis[0].end(); it++)
        {
            if (vis[1][it->first] != it->second)
            {
                flag = 0;
                break;
            }
        }
        if (flag)
            printf("YES\n");
        else
            printf("NO\n");
    }
}