d13r9.cpp

Visual C++ 常用数值算法集 源代码

#include "iostream.h"
#include "stdlib.h"
#include "math.h"

void main()
{
    //program d13r9
    //driver for routine chstwo
    int i,j,npts = 2000;
    int nbins = 10;
    double bins1[11], bins2[11];
    long idum = -17;
    for (j = 1; j<=nbins; j++)
	{
        bins1[j] = 0.0;
		bins2[j] = 0.0;
    }
	double x;
	int ibin;
    for (i = 1; i<=npts; i++)
	{
        x = expdev(idum);
        ibin = int(x * nbins / 3.0) + 1;
        if (ibin <= nbins)
		{
			bins1[ibin] = bins1[ibin] + 1.0;
		}
		x = expdev(idum);
        ibin = int(x * nbins / 3.0) + 1;
        if (ibin <= nbins)
		{
			bins2[ibin] = bins2[ibin] + 1.0;
		}
    }
	double df,chsq,prob;
    chstwo(bins1, bins2, nbins, -1, df, chsq, prob);
    cout<<endl;
    cout<<"    Dataset 1    Dataset 2"<<endl;
	cout.setf(ios::fixed|ios::right);
	cout.precision(2);
    for (i = 1; i<=nbins; i++)
	{
		cout.width(12);
        cout<<bins1[i];
		cout.width(12);
        cout<<bins2[i]<<endl;
    }
    cout<<endl;
	cout.unsetf(ios::fixed);
	cout.setf(ios::scientific);
	cout.precision(6);
    cout<<"chi-squared: "<<chsq<<endl;
    cout<<"probability: "<<prob<<endl;
}