做题思路两个数组一个数组存储每个家庭的人口数，一个数组存储每个家庭分到的房子数
以人数最少的家庭为中心（人数最少的家庭房子数分为1），分为左右两个部分；
左边部分，从数组下标大往小遍历，如果前一个比后一个人数多那么就在后一个房子数量上+1，如果小于等于的话，前一个家庭的房子数为1；
右边部分与左部分相反，数组下标从小往大遍历，如果后一个比前一个家庭人口数多，那么后一个家庭分到的房子数为前一个家庭数+1，否则为1；
注：最小人口的家庭位置可能为端点，即数组下标0或者家庭数n-1，这里需要提前判断出来
#include <iostream>
#include <vector>#include <algorithm>
using namespace std;
int main() {
int n;
cin >> n;
vector <int> v(n, 0);
for (int i = 0; i < n; i++) {
cin >> v[i];
}
int minPosition = min_element(v.begin(), v.end()) - v.begin();//找出最小人数的位置
vector <int> b(n, 0);
b[minPosition] = 1;
if (minPosition > 0) {//分为左右两个部分
for (int i = minPosition-1; i >= 0; i--) {
if (v[i] > v[i + 1])
b[i] = b[i + 1] + 1;
else
b[i] = 1;
}
}
if (minPosition<n - 1) {
for (int i = minPosition + 1; i < n; i++) {
if (v[i] > v[i - 1])
b[i] = b[i - 1] + 1;
else
b[i] = 1;
}
}
int sum = 0;
for (int i = 0; i < n; i++)
sum += b[i];
cout <<sum;
}
全部评论
(0) 回帖