POJ 2847 Widget Factory

POJ_2947

    将M条信息翻译成同余方程组后用高斯消元求解就可以了。

#include<stdio.h>
#include<string.h>
#include<iostream>
#define MAXD 310
using namespace std;
int N, M, mat[MAXD][MAXD], ans[MAXD], h[MAXD];
char a[5], b[5], day[][5] = {"MON", "TUE", "WED", "THU", "FRI", "SAT", "SUN"};
int getday(char *b)
{
    int i;
    for(i = 0; i < 7; i ++)
        if(strcmp(b, day[i]) == 0)
            break;
    return i;
}
void init()
{
    int i, j, k, n;
    for(i = 0; i < M; i ++)
    {
        scanf("%d", &n);
        scanf("%s%s", a, b);
        mat[i][N] = (getday(b) - getday(a) + 8) % 7;
        memset(h, 0, sizeof(h));
        for(j = 0; j < n; j ++)
        {
            scanf("%d", &k);
            ++ h[k - 1];
        }
        for(j = 0; j < N; j ++)
            mat[i][j] = h[j] % 7;
    }
}
int gauss()
{
    int i, j, k, x, y, a, b;
    for(i = j = 0; i < M && j < N; i ++, j ++)
    {
        if(mat[i][j] == 0)
        {
            for(k = i + 1; k < M; k ++)
                if(mat[k][j])
                    break;
            if(k == M)
            {
                -- i;
                continue;
            }
            for(y = j; y <= N; y ++)
                swap(mat[i][y], mat[k][y]);
        }
        for(x = i + 1; x < M; x ++)
            if(mat[x][j])
            {
                a = mat[i][j], b = mat[x][j];
                for(y = j; y <= N; y ++)
                    mat[x][y] = ((mat[x][y] * a - mat[i][y] * b) % 7 + 7) % 7;
            }
    }
    for(k = i; k < M; k ++)
        if(mat[k][N])
            return -1;
    if(i < N)
        return -2;
    for(i = N - 1; i >= 0; i --)
    {
        b = mat[i][N];
        for(j = i + 1; j < N; j ++)
            b -= mat[i][j] * ans[j];
        b = (b % 7 + 7) % 7;
        for(a = 3; a <= 9; a ++)
            if(((a * mat[i][i] % 7 + 7) % 7) == b)
                break;
        ans[i] = a;
    }
    return 0;
}
void solve()
{
    int i, k = gauss();
    if(k == -2)
        printf("Multiple solutions.\n");
    else if(k == -1)
        printf("Inconsistent data.\n");
    else
    {
        printf("%d", ans[0]);
        for(i = 1; i < N; i ++)
            printf(" %d", ans[i]);
        printf("\n");
    }
}
int main()
{
    for(;;)
    {
        scanf("%d%d", &N, &M);
        if(!N && !M)
            break;
        init();
        solve();
    }
    return 0;
}