题面:
题意:
一排 n 个位置,部分位置上有一定数量的砖头,每次操作可选择一对 i 和 j,将 i 处的砖头搬一块到 j 处,其中,要求 i < j 且区间 [ i+1 , j-1 ] 的位置上都得有砖头。问最少需要多少次操作才能将所有砖头都堆叠在 n 处(即最后一个位置上)。
题解:
一步步去模拟明显不行,基于贪心的思想,如果对于每一块砖头都能做到一次操作就搬到 n 处,自然是最好。但前提是某一块砖头被搬起的地方 i 到 n 之间的所有位置上都得有砖头。所以对于本来就有砖头的位置,我们无需考虑(在我们架设好一路都有砖头的情况下,有几块砖头对应得花几次操作),而本来就没有砖头的位置,我们都得花一次操作给它添加一个砖头。最终,问题就转化为:从最开始有砖头的位置往后扫,累加每次给空位置添加一块新砖头的操作次数和原本就需要搬的砖头数。
代码:
#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
int T,n;
signed main()
{
std::ios::sync_with_stdio(false);
std::cin.tie(0);
cin>>T;
while(T--)
{
cin>>n;
vector<int> a(n+1);
for(int i=1;i<=n;i++) cin>>a[i];
int p=1,ans=0;
while(p<=n&&!a[p]) p++;//找初始有砖头的位置
for(int i=p;i<n;i++)
{
ans+=a[i];
if(!a[i]) ans++;//添加新砖头
}
cout<<ans<<endl;
}
return 0;
}