結果

問題 No.1007 コイン集め
ユーザー ate
提出日時 2020-07-08 22:30:18
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 62 ms / 1,500 ms
コード長 592 bytes
コンパイル時間 614 ms
コンパイル使用メモリ 73,352 KB
最終ジャッジ日時 2025-01-11 17:04:09
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
signed main(){
  int n,k;
  std::cin>>n>>k;
  k--;
  std::vector<int> a(n);
  for(auto& ai:a)std::cin>>ai;
  int l=k-1,r=k+1;
  while(l>0){
    if(a[l]<=1)break;
    l--;
  }
  while(r<n){
    if(a[r]<=1)break;
    r++;
  }
  if(a[k]==0){
    std::cout<< 0 <<std::endl;
    return 0;
  }
  if(a[k]==1){
    int64_t suml=0,sumr=0;
    for(int i=k;i>=l;--i)suml+=a[i];
    for(int i=k;i<=r;++i)sumr+=a[i];
    std::cout<< std::max(suml,sumr) <<std::endl;
    return 0;
  }
  int64_t ans = 0;
  for(int i=l;i<=r;++i)ans+=a[i];
  std::cout<< ans <<std::endl;
}
0