class Solution {
public:
int candy(vector<int> &ratings) {
int len = ratings.size();
vector<int> result(len, 1);
int inc = 1, count = 0;
for(int i = 0; i < len; ++i){
if(i >= 1 && ratings[i] > ratings[i-1]){
result[i] = max(++inc, result[i-1]+1, 0);
}else{
inc = 1;
}
}
for(int j = len-1, inc = 1; j >= 0; --j){
if(j < len-1 && ratings[j] > ratings[j+1]){
result[j] = max(++inc, result[j+1]+1, result[j], 0);
}else{
inc = 1;
}
}
for(int i = 0; i < len; i++)
count += result[i];
return count;
}
private:
int max(int x, int y, ...){
va_list arg_ptr;
int nArgValue = 0;
int maxValue = x>y?x:y;
va_start(arg_ptr, y);//va_start函数将参数arg_ptr设置为可变参数列表的第一个参数
while(nArgValue != 0)//自定义结束条件是输入参数为0
{
nArgValue=va_arg(arg_ptr,int);
maxValue = maxValue>nArgValue?maxValue:nArgValue;
}
va_end(arg_ptr);
return maxValue;
}
};
注:max函数未定义,会出现required from here 的错误。
以下为转载:
C++允许定义形参个数和类型不确定的函数。例如,C语言中的标准函数printf便使用这种机制。在声明不确定形参的函数时,形参部分可以使用省略号“…”代替。“…”告诉编译器,在函数调用时不检查形参类型是否与实参类型相同,也不检查参数个数。
例如:void ConnectData(int i,...) 在上面的代码中,编译器只检查第一个参数是否为整型,而不对其他参数进行检查。
对于可变参数的函数,需要进行特殊的处理。首先需要引用 <stdarg.h>头文件,然后利用va_list类型和va_start、va_arg、va_end 3个宏读取传递到函数中的参数值。
这几个宏的定义如下(在 ANSI C 中):
type va_arg( va_list arg_ptr, type );void va_end( va_list arg_ptr );void va_start( va_list arg_ptr, prev_param );
说明如下:
va_start :sets arg_ptr to the first optional argument in the list of arguments passed to the function. The argument arg_ptr must have va_list type. The argument prev_param is the name of the required parameter immediately preceding the first optional argument in the argument list. If prev_param is declared with the register storage class, the macro’s behavior is undefined. va_start must be used before va_arg is used for the first time.
【 va_start函数将参数arg_ptr设置为可变参数列表的第一个参数。参数arg_ptr的类型必须为va_list。参数prev_param是在可变参数列表之前的那一个参数。(也就是说在 ANSI C 中,如果一个函数有可变参数,那么在该可变参数前必须有一个明确定义的参数,否则无法调用函数 va_start ,例如函数 int add(int i,...)是合法的,而函数 int add(...)是不合法的。)】
va_arg :retrieves a value of type from the location given by arg_ptr and increments arg_ptr to point to the next argument in the list, using the size of type to determine where the next argument starts. va_arg can be used any number of times within the function to retrieve arguments from the list.
【 va_arg函数将返回 arg_ptr 所指位置的值,并将 arg_ptr 指向下一个参数 】
va_end
After all arguments have been retrieved, va_end resets the pointer to NULL.