华为2018年8月29日笔试第三题

今晚刚做完华为笔试的三个题目，先将第三题的代码，展现出来，希望互相学习，题目的大致内容是关于牛生牛的问题。

#include<iostream>
using namespace std;
int sum(int niu, int yue);
int main(){
	int a,n;
	cin>>a;
	n = 2*a;
	int b[n];
	int i,j;
	for (i=0;i<n;i++){
		cin>>b[i];
	}
	for (j = 0; j<n; j++){
		cout<<sum(b[j],b[j+1])<<endl;
		j++;
	}
	return 0;
}
int sum(int niu, int yue)
{
	int i,num = 0;
	for (i=1;0<(yue-(i+3));i++){
		num += yue-(i+3);
	}
	num = niu*(yue + num +1);
	return num;	
}

具体的题目，暂时想不起来了，网上似乎也没有，等题目出来了，我再粘贴过来。