글 읽기 - 왜안되는지 모르겠습니다

sontg123 3년 전 0

인터넷에 있는 rref예제들도 다 되고 테스트케이스도 다 돌아갑니다 왜 틀렸는지 모르겠습니다

입력은 분모 분자 모두 1자리수라고 가정하고 했습니다 분모나 분자가 [-9,9] 이므로 2자리수 넘어가면 안되긴 합니다..

답답해죽겠습니다 맞왜틀.,.

#include <iostream>
#include <vector>
using namespace std;
typedef vector<vector<pair<int,int>>> matrix;
int gcd(int a,int b){
    if(a==0)
        return b;
    b=b%a;
    if(b==0)
        return a;
    return gcd(b,a);
};
class RREF{
private:
    matrix mt;
    int N,M;
public:
    RREF(){
        std::cin >> N >> M;
        mt.resize(N);
        set_val();
    }
    void set_val(){
        string str;
        for(int i=0;i<N;i++){
            for(int j=0;j<M;j++){
                std::cin >>str;
                if(str[0]=='-'){
                    if(str.size()==2){
                        mt[i].push_back({-(str[1]-'0'),1});
                    }
                    else{
                        mt[i].push_back({-(str[1]-'0'),str[3]-'0'});
                    }
                }
                else{
                    if(str.size()==1){
                        mt[i].push_back({str[0]-'0',1});
                    }
                    else{
                        mt[i].push_back({str[0]-'0',str[2]-'0'});
                    }
                }
            }
        }
    }
    void print(){
        for(int i=0;i<N;i++){
            for(int j=0;j<M;j++){
                if(mt[i][j].second==1)
                    std::cout << mt[i][j].first <<' ';
                else
                    std::cout << mt[i][j].first<<'/'<<mt[i][j].second <<' ';
            }
            std::cout << '\n';
        }
    }
    pair<int,int> p_mi(pair<int,int> a,pair<int,int> b){
        int u1=a.first*b.second,d1=a.second*b.second;
        int u2=b.first*a.second,d2=b.second*a.second;
        int g=gcd(abs(u1-u2),d1);
        return {(u1-u2)/g,d1/g};
    }
    pair<int,int> p_mul(pair<int,int> a,pair<int,int> b){
        if(a.second<0){
            a.first=-a.first;
            a.second=-a.second;
        }
        int g=gcd(abs(a.first*b.first),a.second*b.second);
        return {a.first*b.first/g,a.second*b.second/g};
    }
    void gaussian_elimination(){
        vector<pair<int,int>> pivot;
        pair<int,int> T;
        int tmp=0;
        bool flag=false;
        int cur_i,cur_j;
        for(int j=0;j<M;j++){
            for(int i=tmp;i<N;i++){
                if(mt[i][j].first!=0){
                    pivot.push_back({tmp,j});
                    cur_i=tmp,cur_j=j;
                    swap(mt[tmp++],mt[i]);
                    flag=true;
                    break;
                }
            }
            if(flag){
                for(int i=cur_i+1;i<N;i++){
                    if(mt[i][cur_j].first!=0){
                        T.first=mt[i][cur_j].first*mt[cur_i][cur_j].second;
                        T.second=mt[cur_i][cur_j].first*mt[i][cur_j].second;
                        for(int l=0;l<M;l++){
                            mt[i][l]=p_mi(mt[i][l],p_mul(T,mt[cur_i][l]));
                        }
                    }
                }
                flag=false;
            }
            if(tmp==N)
                break;
        }
        pair<int,int> T2;
        for(int i=pivot.size()-1;i>=0;i--){
            T2=mt[pivot[i].first][pivot[i].second];
            T.first=T2.second;
            T.second=T2.first;
            if(T2.first!=1){
                for(int j=pivot[i].second;j<M;j++){
                    mt[pivot[i].first][j]= p_mul(T,mt[pivot[i].first][j]);
                }
            }
            for(int j=pivot[i].first-1;j>=0;j--){
                if(mt[j][pivot[i].second].first!=0){
                    T.first=mt[j][pivot[i].second].first;
                    T.second=mt[j][pivot[i].second].second;
                    for(int l=pivot[i].second;l<M;l++){
                        mt[j][l]=p_mi(mt[j][l],p_mul(T,mt[pivot[i].first][l]));
                    }
                }
            }
        }
        print();
        
    }

};
int main(){
    RREF r;
    r.gaussian_elimination();

    return 0;
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

왜안되는지 모르겠습니다

sontg123 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말