結果
問題 | No.1007 コイン集め |
ユーザー |
![]() |
提出日時 | 2020-03-06 22:09:33 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 13 ms / 1,500 ms |
コード長 | 770 bytes |
コンパイル時間 | 1,439 ms |
コンパイル使用メモリ | 168,212 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-14 07:13:30 |
合計ジャッジ時間 | 2,369 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
#include <bits/stdc++.h> #define all(v) v.begin(),v.end() #define eb emplace_back #define N 1000010 using namespace std; using ll=long long; using P=pair<int,int>; const ll INF=1LL<<30; inline void chmin(int &a,int b){a=min(a,b);} int dy[4]={0,-1,0,1}; int dx[4]={1,0,-1,0}; int to[4]={2,3,0,1}; int main(){ ios::sync_with_stdio(0); cin.tie(0); int n,k;cin>>n>>k;--k; vector<ll> a(n); for(int i=0;i<n;i++){ cin>>a[i]; } ll l=0,r=0; for(int i=k-1;i>=0;i--){ if(a[i]==0)break; if(a[i]==1){ l++; break; }else{ l+=a[i]; } } for(int i=k+1;i<n;i++){ if(a[i]==0)break; if(a[i]==1){ r++; break; }else{ r+=a[i]; } } if(a[k]==0){ cout<<0<<endl; }else if(a[k]==1){ cout<<max(l,r)+1LL<<endl; }else{ cout<<l+r+a[k]<<endl; } }