1 Star 0 Fork 1

saigon/Algorithms

forked from charlieshu/Algorithms 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
U321334 ╙ы╖╟║═.cpp 709 Bytes
一键复制 编辑 原始数据 按行查看 历史
charlie 提交于 2024-01-09 00:01 . move from github to gitee
#include <iostream>
using namespace std;
int dp[200005];
int a[200005];
int main(){
int n;
cin>>n;
for(int i=0;i<n;i++){
cin>>a[i];
// dp[i][i] = a[i];
}
long long ans=0;
ans += a[0];
dp[0] = a[0];
for(int i=1;i<n;i++){
// cout<<i<<" "<<ans<<endl;
if(a[i] == 0){
ans += i;
}
else{
dp[i] = dp[i-1]+1;
if(dp[i]%2 == 1){
if(i-dp[i] >= 0 && a[i-dp[i]] == 0)
ans += dp[i]/2+2;
else
ans += dp[i]/2+1;
}
else{
ans += i-dp[i]>0?i-dp[i]:0;
ans += dp[i]/2;
}
}
}
// for(int i=0;i<n;i++){
// for(int j=0;j<n;j++){
// cout<<dp[i][j]<<" ";
// }
// cout<<endl;
// }
cout<<ans;
return 0;
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/saigonshu/algorithm.git
git@gitee.com:saigonshu/algorithm.git
saigonshu
algorithm
Algorithms
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385