电子协会 C语言 1级 18 、计算邮资

http://noi.openjudge.cn/ch0104/14/

C++代码：

#include 
using namespace std;
int main()
{
	//sum=8 
	//重量在1000克以内(包括1000克), 基本费8元。
	//这样写比较简单 
	int a,sum=8;
	char b;
	
	scanf("%d %c",&a,&b);
	
	if(a>1000) 
	{
		{
			if((a-1000)%500==0) 
			{
				sum+=(a-1000)/500*4;
			}
			else
			{
				sum+=((a-1000)/500+1)*4;
			}
		}
	}	
	
	if(b=='y')
	{
		sum+=5;
	}
	
	printf("%d",sum);
	
	return 0;
}

import math
x, y = map(str, input().split())
if y == "y":
    if int(x) <= 1000:
        print(13)
    else:
        print(8 + math.ceil((int(x)-1000)/500)*4 + 5)
if y == "n":
    if int(x) <= 1000:
        print(8)
    else:
        print(8 + math.ceil((int(x)-1000)/500)*4)