poj 2947 Widget Factory

这几把题意我感觉很变态啊，题意：https://blog.csdn.net/qingshui23/article/details/52511894
#include <iostream>// 用G++，还有就是不能外挂
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <map>
//#define IO  ios::sync_with_stdio(false),cin.tie(0), cout.tie(0);
//#pragma comment(linker, "/STACK:1024000000,1024000000")
void ex_gcd(int a, int b, int &d, int &x, int &y) { if (!b) { x = 1; y = 0; d = a; } else { ex_gcd(b, a%b, d, y, x); y -= x * (a / b); }; }
int gcd(int a, int b) { return b ? gcd(b, a%b) : a; }
int lcm(int a,int b){return a/gcd(a,b)*b;}//先除后乘防溢出
int inv_exgcd(int a, int m) { int d, x, y;ex_gcd(a, m, d, x, y);return d == 1 ? (x + m) % m : -1; }
typedef long long ll;
const int maxn=1e3;
using namespace std;
int a[maxn][maxn];
int x[maxn];
int n,m;
map<string,int>Mon;
bool free_x[maxn];//记录不确定的变元
void IN()
{
    Mon["MON"]=1,Mon["TUE"]=2,Mon["WED"]=3;
    Mon["THU"]=4,Mon["FRI"]=5,Mon["SAT"]=6,
    Mon["SUN"]=7;
}
int Gauss(int equ,int var)
{
    int max_r,col,k;
    int ta,tb;
    int Lcm;
    int temp;
    int free_x_num=0,free_index;
    for(int i=0;i<=var;++i) x[i]=0,free_x[i]=true;
    //转化为阶梯例
    col=0;//当前列
    for( k=0,col=0;k<equ&&col<var;++k,++col)
    {//枚举当前行
       max_r=k;
       for(int i=k+1;i<equ;++i)
            if(abs(a[i][col])>abs(a[max_r][col]))max_r=i;
       if(max_r!=k)
           for(int i=k;i<=var;++i) swap(a[k][i],a[max_r][i]);///i=0;
        if(!a[k][col])
        {
            k--;
            continue;//处理下一列
        }
      for(int i=k+1;i<equ;++i)
      {//化为阶梯型
          if(a[i][col])
          {
              Lcm=lcm(abs(a[i][col]),abs(a[k][col]));
              ta=Lcm/abs(a[i][col]),tb=Lcm/abs(a[k][col]);
              if(a[i][col]*a[k][col]<0)tb=-tb;
              for(int j=col;j<=var;++j)
              {
                  a[i][j]=((a[i][j]*ta-a[k][j]*tb)%7+7)%7;
              }
          }
      }
    }
    for(int i=k;i<equ;++i)// 无解
        if(a[i][col]) return -1;
    if(k<var)// 无穷解
    {///这这儿可以直接 return var-k;不用求出未知元
        for(int i=k-1;i>=0;--i)
        {
            free_x_num=0;//用于判断不确定性变元的数量，若超过一个仍然无法求解(无解)
            for(int j=0;j<var;++j)
                if(a[i][j]&&free_x[j])free_x_num++,free_index=j;
            if(free_x_num>1)continue;
            temp=a[i][var];
            for(int j=0;j<var;++j)
            {
                if(a[i][j]&&j!=free_x_num)temp=((temp-a[i][j]*x[j])%7+7)%7;;
            }
            x[free_index]=(temp/a[i][free_index])%7;
            free_x[free_index]=false;
        }
        return var-k;
    }
    //唯一解
    for(int i=var-1;i>=0;--i)
    {
        temp=a[i][var]%7;
        for(int j=i+1;j<var;++j)
            if(a[i][j])temp=((temp-a[i][j]*x[j])%7+7)%7;//cout<<temp<<endl;
         // if(temp%a[i][i])return -2;//有浮点数解， 说明下一行的temp/a[i][i]不是整数,也是有道理的
         // x[i]=(temp/a[i][i])%7;
        int d=inv_exgcd(a[i][i],7); if(d==-1)return -2; ///当逆元=-1是也是无整数解（求逆元又必须是互质的情况下）
        x[i]=temp*d%7;
        if(x[i]<3)x[i]+=7;
        if(x[i]>9)x[i]%=7;
    }
    return 0;
}
void init()
{
    memset(a,0,sizeof(a));
    int p,equ=m,var=n,pi;
    char s[10],e[10];
    for(int i=0;i<m;++i)
    {
        scanf("%d%s %s",&p,s,e);
        a[i][var]=(Mon[e]-Mon[s]+1+7)%7;a[i][var]=(a[i][var]+7)%7;
        for(int j=0;j<p;++j)
        {
            scanf("%d",&pi);
            a[i][pi-1]=(a[i][pi-1]+1)%7;
        }
    }
}
int main()
{//IO;
    IN();
    while(scanf("%d%d",&n,&m)&&(n||m))
    {
        init();
        int free_num=Gauss(m,n);
        if(free_num==-1)printf("Inconsistent data.\n");
        else if(free_num==-2)printf("有浮点数解无整数解");
        else if(free_num>0)printf("Multiple solutions.\n");
        else
        {
            printf("%d",x[0]);
            for(int i=1;i<n;++i)
                printf(" %d",x[i]);
            puts("");
        }
    }
}
poj 2947 Widget Factory

猜你喜欢
