标签:
题意:有n件装饰品,有m组信息。(1 <= n ,m<= 300)每组信息有开始的星期和结束的星期(是在mod 7范围内的)并且还包括num种装饰品的种类(1~n),其中每种装饰品所用的时间3 <= x[i] <= 9;种类的输入可以重复;
思路:
1.根据输入建立增广矩阵a[][],但是在建立和求解的过程中由于是mod意义下的,所以输入的个数和最终所用的时间都要mod 7;(分析可知当个数是7的同余类时,开始星期相同则结束星期也相同)
2.前面几个高斯消元,我用的是free_var来判断是否有自由变元,这是在输入的方程数和求解变元数相等的情况才成立。在本题中对于sample 1就会发现方程数原本就比变元多1,这时计算出的free_var = 1,但是并不是将就有了一个维度的自由变元。还是要看有用方程的个数row与var之间的关系;
3.在得到上三角阵求解变元x[i]的时候,需要解一个模线性方程,a[i][i]*x[i] + 7*y = ret(mod 7);ret为a[i][col]用已知的x[j]消去除a[i][i]得到的;
这时调用exgcd()即可求解;最后注意解要在3~9范围内即可;
ps:时间性能不是很好,竟然用了1782ms...最短的是297ms..差距啊!!!
#include<iostream> #include<cstdio> #include<cstring> #include<string.h> #include<algorithm> #include<map> #include<queue> #include<vector> #include<cmath> #include<stdlib.h> #include<time.h> using namespace std; #define rep0(i,l,r) for(int i = (l);i < (r);i++) #define rep1(i,l,r) for(int i = (l);i <= (r);i++) #define rep_0(i,r,l) for(int i = (r);i > (l);i--) #define rep_1(i,r,l) for(int i = (r);i >= (l);i--) #define MS0(a) memset(a,0,sizeof(a)) #define MS1(a) memset(a,-1,sizeof(a)) int a[305][305]; int equ,var; int x[305]; const int MOD = 7; void debug() { puts("********"); int i,j; rep0(i,0,equ){ rep1(j,0,var) cout<<a[i][j]<<" "; cout<<endl; }puts("********"); } 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; } void exgcd(int a,int b,int& d,int& x,int& y) { if(!b){d = a;x = 1;y = 0;} else{ exgcd(b,a%b,d,y,x); y -= x*(a/b); } } int Guass() { int i,j,k,free_var = 0,row,col; for(row = 0,col = 0;row < equ && col < var;row++,col++){ int mx = row; rep0(j,row+1,equ) if(abs(a[j][col]) > abs(a[mx][col])) mx = j; if(a[mx][col] == 0){ row--; // 行不变;不能通过这里记录自由变元的个数,只能记录没用的col continue; } if(mx != row) rep1(k,col,var) swap(a[row][k],a[mx][k]); rep0(j,row+1,equ){ if(a[j][col]){ int lcm = LCM(abs(a[row][col]),abs(a[j][col])); int ration_row = lcm/a[row][col],ration_j = lcm/a[j][col]; if(a[row][col]*a[j][col] < 0) ration_row = -ration_row; //符号相反变加法; rep1(k,col,var) a[j][k] = ((a[j][k]*ration_j - a[row][k]*ration_row)%7+7)%7; } } } //debug(); rep0(i,row,equ) if(a[i][var] != 0) return -1; //无解 if(row < var) return var - row;//row表示有用的方程数方程,但是要在判断出有解的前提下才能说有多组解; rep_1(i,var - 1,0){ // ***若为唯一解,其实就是var维方阵 int ret = a[i][var]; for(j = i+1;j < var;j++) //利用已求得的变元消去第row行col后面的元素,得到一元方程; ret -= x[j]*a[i][j]; ret = ((ret%7)+7)%7; int d,x1,y; //构造出 a[i][i]*x[i] + 7*y = ret(mod 7);且gcd(a[row][col],7) = 1)因为a[row][col] != 0 exgcd(a[i][i],7,d,x1,y); //之后乘上ret弄到3~9范围即可; x[i] = ((ret*x1)%7+7)%7; if(x[i] < 3) x[i] += 7; } return 0; } const char str[7][5] = {{"MON"},{"TUE"},{"WED"},{"THU"},{"FRI"},{"SAT"},{"SUN"}}; int date_id(char *c) { for(int i = 0;i < 7;i++) if(strcmp(str[i],c) == 0) return i; } int main() { int i,j,n,m; char s[5],t[5]; while(scanf("%d%d",&n,&m) == 2 && n + m){ MS0(a); equ = m;var = n; int kind,num; rep0(i,0,m){ scanf("%d%s%s",&num,s,t); a[i][var] = date_id(t)-date_id(s)+1; if(a[i][var] < 0) a[i][var] += 7; rep0(j,0,num){ scanf("%d",&kind); a[i][--kind]++; } rep1(j,0,var) a[i][j] %= 7; } //debug(); int ret = Guass(); if(ret == -1) puts("Inconsistent data."); else if(ret > 0) puts("Multiple solutions."); else{ rep0(i,0,var) printf("%d%c",x[i],i == var - 1?‘\n‘:‘ ‘); } } return 0; }
标签:
原文地址:http://www.cnblogs.com/hxer/p/5180813.html