P6056 [加油武汉]SIR 模型

题目背景
SIR 模型将总人口分为以下三类：

易感者(susceptibles)，其数量记为 s(t) ，表示 t 时刻未染病但有可能被该类疾病传染的人数；
染病者(infectives)，其数量记为 i(t)，表示 t 时刻已被感染成为病人而且具有传染力的人数；
恢复者(recovered)，其数量记为 r(t)，表示 t 时刻已从染病者中移出的人数。
设总人口为N(t)，则有)N(t)=s(t)+i(t)+r(t)。

SIR模型的建立基于以下三个假设：

不考虑人口的出生、死亡、流动等种群动力因素。人口始终保持一个常数，即 N(t) \equiv KN(t)≡K。
一个病人一旦与易感者接触就必然具有一定的传染力。假设 tt 时刻单位时间内，一个病人能传染的易感者数目与此环境内易感者总数 s(t)s(t) 成正比，比例系数为 \betaβ，从而在t时刻单位时间内被所有病人传染的人数为 \beta s(t)i(t)βs(t)i(t)。
tt 时刻，单位时间内从染病者中移出的人数与病人数量成正比，比例系数为\gammaγ，单位时间内移出者的数量为 \gamma i(t)γi(t)。
题目描述
我们将这个模型简化一下，初始有感染者 I 人和易感者 S 人，对于每一天当前有 Ii个感染者，Si个易感者，Ri个恢复者，则每天会有 ceil⌈βSiIi⌉ 人被感染（由易感者变成感染者），有ceil⌈γIi⌉ 人被治愈（由感染者变成恢复者） 。
其中 β 为感染系数γ 为恢复系数 ceil⌈⌉ 为上取整符号。

求 n 天后，有多少易感者 S，感染者 I，和恢复者 RR。

注: 感染者和恢复者都是每天结算的，结算的结果只和当天开始的时候的值有关，即感染者当天恢复不影响他当天感染别人。

若计算被感染人数超过易感者人数则全员被感染。

输入格式
第一行三个正整数，分别表示第 0 天易感者人数 S0和感染者人数I0，以及天数 n（刚开始恢复者数 R0 =0）。

第二行两个浮点数，分别表示感染系数 β 和恢复系数 γ。

输出格式
一行三个整数，分别表示 n 天后的易感者人数 S 、感染者人数 I 和恢复者R。

输入输出样例
输入 #1

980 20 2

0.0005 0.00001

输出 #1

955 43 2

输入 #2

1400000000 1 10

0.000000003 0.001

输出 #2

1386791252 13205592 3157

输入 #3

1919 810 1
0.00001 0.1

输出 #3

1903 745 81

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#include<math.h>
using namespace std;

int s , i, n,r;
double a,b;
int main()
{
    
    cin>>s>>i>>n;
    cin>>a>>b;
    for(int j=0;j<n;j++)
    {
        int Si = ceil(a*s*i);
        int Ii = ceil(b*i);
        r += Ii;
        if(Si >= s) {
            i+=s - Ii;
            s = 0;
        } else{
            i+=Si - Ii;
            s -= Si;
        }
        
	}
	cout<<s<<" "<<i<<" "<<r;
    return 0;
}