結果
問題 | No.1765 While Shining |
ユーザー |
![]() |
提出日時 | 2021-11-26 22:26:36 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 42 ms / 2,000 ms |
コード長 | 533 bytes |
コンパイル時間 | 5,201 ms |
コンパイル使用メモリ | 249,480 KB |
最終ジャッジ日時 | 2025-01-26 01:27:26 |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint998244353; using namespace std; #define rep(i,n) for(int i=0;i<(n);i++) #define Inf 1000000001 int main(){ int N; cin>>N; vector<int> a(N); rep(i,N)cin>>a[i]; vector<int> dp(N,0); dp.back() = N-1; for(int i=N-2;i>=0;i--){ if(a[i+1]!=a[i]){ dp[i] = dp[i+1]; } else{ dp[i] = i+1; } } long long ans= 0; rep(i,N){ if(a[i]==0)continue; ans += dp[i] - i; } cout<<ans<<endl; return 0; }