런타임 오류가 나옵니다 roots문제

  • kksuhj
    kksuhj 
    RTE (SIGSEGV: segmentation fault, probably incorrect memory access)

    라는 오류가 나옵니다.

    roots문제를 푸는 중에 나왔는데요

    visual studio에서 돌릴때는 아무 이상이 없던 것이

    채점시에 문제가 되네요

    코드는 아래와 같습니다.. 

    #include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <vector>
#include <math.h>
#include <algorithm>
using namespace std;

int testcase,n;
vector<double> poly;
const double limit=10;

void input()
{
    vector<double> empty;
    double temp;

    poly=empty;

    cin>>n;
    for(int i=0;i<n+1;i++)
    {
        cin>>temp;
        poly.push_back(temp);
    }   
}

vector<double> differentiate(const vector<double>& p)
{
    vector<double> dif;
    for(int i=0;i<n;i++)
        dif.push_back( (n-i)*p[i] );
    return dif;
}

vector<double> solveNaive(const vector<double>& p)
{
    vector<double> ret;

    //1차식
    if(p.size()==2)
    {
        ret.push_back(-p[1]/p[0]);
    }else if(p.size()==3)//2차식
    {
        double a=p[0],b=p[1],c=p[2];
        ret.push_back( (-b+sqrt(pow(b,2)-4*a*c))/(2*a) );
        ret.push_back( (-b-sqrt(pow(b,2)-4*a*c))/(2*a) );
    }
    sort(ret.begin(),ret.end());
    return ret;
}

double eval(const vector<double>& p, double x)
{
    double ret=0;
    for(int i=0; i<=n ;i++)
        ret+=pow(x,n-i)*p[i];
    return ret;
}

vector<double> solve(const vector<double>& p)
{
    int degree=p.size()-1;

    if(degree<=2) return solveNaive(p);
    vector<double> ret;
    vector<double> derivative = differentiate(p);
    vector<double> sols = solve(derivative);

    sols.insert(sols.begin(),-limit-1);
    sols.insert(sols.end(),limit+1);

    for(int i=0;i<sols.size()-1;i++)
    {
        double x1=sols[i],x2=sols[i+1];
        double y1=eval(p,x1),y2=eval(p,x2);

        if(y1*y2>0) continue;
        if(y1>y2) {swap(y1,y2); swap(x1,x2);}

        for(int i=0;i<100;i++)
        {
            double mx=(x1+x2)/2;
            double my=eval(p,mx);

            if(y1*my<=0)
            {
                x2=mx;
                y2=my;
            }else
            {
                x1=mx;
                y1=my;
            }
        }
        ret.push_back( (x1+x2)/2 );
    }
    sort(ret.begin(),ret.end());
    return ret;
}

int main()
{
    cin>>testcase;
    while(testcase--)
    {
        input();
        vector<double> ret(solve(poly));// 이 부분에서 문제인듯 합니다. 
        for(int i=0;i<ret.size();i++)
            printf("%.8f",ret[i]);
    }
    return 0;
}

    마지막에 메인에서 vector ret(solve(poly));에 문제가 있는 듯 한데 어떤식으로 고쳐야 할 지를 잘 모르겠습니다... 도움을 주세요 ㅠㅠ


    10년 전
2개의 댓글이 있습니다.
  • Signin
    Signin

    함수 differentiate()과 eval()에서,
    반복문 도는 횟수가 항상 n으로 고정되어 문제가 생기는 것 같습니다.

    처음 인풋이 5차식으로 들어와도, 극대/극소 위치를 구하기 위해서
    식을 미분하다 보면, 차수가 점점 줄어들게 됩니다.

    이 경우에 주어진 식은 3차식(원소 4개인 vector)인데
    반복문은 5번 돌게 되고, 따라서 존재하지 않는 p[4]를 참조하게 됩니다.

    각 함수의 n을 p.size()-1로 바꿔주시면 에러가 나지 않을 것 같습니다.


    10년 전 link
  • kksuhj
    kksuhj

    감사합니다. ㅜㅜ 혼자서 터득해야 하는데 이렇게 힘을 빌리네요 ㅠㅠ


    10년 전 link

  • 정회원 권한이 있어야 커멘트를 다실 수 있습니다. 정회원이 되시려면 온라인 저지에서 5문제 이상을 푸시고, 가입 후 7일 이상이 지나셔야 합니다. 현재 문제를 푸셨습니다.