新冠传播模拟

STAR

[C++] 纯文本查看 复制代码

#include<bits/stdc++.h>
#include <graphics.h>             //包含EGE的头文件
#include<time.h>
using namespace std;
int width=800;
int hight=600;
int number;
struct ballon
{
    int r;
    float vx;
    float vy;
    float x;
    float y;
    float g;
    COLORS color;
    int healthy;
}ball[1000];
float inx(int min,int max)
{
    float t=rand()%(max-min)+min;
    return t;
}
void is_ill()
{
	for(int i=0;i<number;i++)
    {
    	for(int j=0;j<number;j++)
    	{
    		abs(ball[i].x-ball[j].x);
    		abs(ball[i].y-ball[j].y);
    		if(((abs(ball[i].x-ball[j].x)<=(ball[i].r+ball[j].r))&&(abs(ball[i].y-ball[j].y)<=(ball[i].r+ball[j].r)))&&(ball[i].healthy!=ball[j].healthy))
    		{
    			ball[i].healthy=1;
    			ball[j].healthy=1;
			}
		}
	}
}
int main()
{
    srand(time(0));//随机种子函数   
    cout<<"请输入病人数量（不少于1000）:";
	cin>>number;  
    for(int i=0;i<number;i++)
    {
        ball[i].x = inx(ball[i].r*2+1,width-ball[i].r);
        ball[i].y = inx(ball[i].r*2+1,hight-ball[i].r);
        ball[i].vx = rand()%3;
        ball[i].vy = rand()%3;
        ball[i].r = 10+rand()%10;
        ball[i].g=0;
        ball[i].color=GREEN;
        ball[i].healthy=0;
    }
    
    ball[0].color=RED;
    ball[0].healthy=1;
    initgraph(width, hight);                //初始化图形界面
    setcolor(RED);  //设置画图线为红色 
    setbkcolor(WHITE);//设置背景颜色为白色
    setfillcolor(RED);//设置填充颜色为红色
    while(1)
    {
		is_ill();
        for(int i=0;i<number;i++)
        {
	        ball[i].vy=ball[i].vy+ball[i].g;
	        ball[i].y=ball[i].vy+ball[i].y;
	        ball[i].vx=ball[i].vx+ball[i].g;
	        ball[i].x=ball[i].vx+ball[i].x;
	        if(ball[i].y>=hight-ball[i].r)
	        {
	            ball[i].y=hight-ball[i].r;
	            ball[i].vy=-ball[i].vy;
	        }
	        if(ball[i].y<=ball[i].r)
	        {
	        	ball[i].y=ball[i].r;
	        	ball[i].vy=-ball[i].vy;
			}
	        if(ball[i].x>=width-ball[i].r)
	        {
	            ball[i].x=width-ball[i].r;
	            ball[i].vx=-ball[i].vx;
	        }
	        if(ball[i].x<=ball[i].r)
	        {
	        	ball[i].x=ball[i].r;
	        	ball[i].vx=-ball[i].vx;
			}
	        fillellipse(ball[i].x, ball[i].y, ball[i].r, ball[i].r); //画一个实心圆 
	        //floodfillsurface(400, 300, RED);
	        setfillcolor(ball[i].color);
	 		
	 		if(ball[i].healthy==1)
	 		{
	 			ball[i].color=RED;
			 }
    	}
    	Sleep(10);
	    cleardevice();
    }
      
    getch();                            //暂停，等待键盘按键
   
    closegraph();        //关闭图形界面
     
    return 0;
}