#440142 (Java21) No.1007 コイン集め

• 問題: No.1007 コイン集め
• ユーザー: 37zigen
• 提出日時: 2020-03-06 23:15:48
• 言語: Java21 (openjdk 21)
• 結果: AC
• 実行時間: 611 ms / 1,500 ms
• コード長: 871 bytes
• コンパイル時間: 1,999 ms
• コンパイル使用メモリ: 73,952 KB
• 実行使用メモリ: 61,004 KB
• 最終ジャッジ日時: 2023-08-04 12:51:50
• 合計ジャッジ時間: 10,242 ms
• ジャッジサーバーID （参考情報）: judge14 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	122 ms; 55,900 KB
testcase_01	AC	121 ms; 55,684 KB
testcase_02	AC	122 ms; 55,672 KB
testcase_03	AC	120 ms; 55,532 KB
testcase_04	AC	120 ms; 55,748 KB
testcase_05	AC	122 ms; 55,636 KB
testcase_06	AC	123 ms; 55,740 KB
testcase_07	AC	124 ms; 55,748 KB
testcase_08	AC	123 ms; 55,752 KB
testcase_09	AC	127 ms; 55,900 KB
testcase_10	AC	126 ms; 53,944 KB
testcase_11	AC	126 ms; 55,684 KB
testcase_12	AC	611 ms; 60,808 KB
testcase_13	AC	597 ms; 60,388 KB
testcase_14	AC	575 ms; 60,700 KB
testcase_15	AC	459 ms; 60,196 KB
testcase_16	AC	477 ms; 60,516 KB
testcase_17	AC	472 ms; 60,256 KB
testcase_18	AC	574 ms; 60,656 KB
testcase_19	AC	573 ms; 60,956 KB
testcase_20	AC	576 ms; 60,880 KB
testcase_21	AC	574 ms; 61,004 KB

ソースコード

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Scanner;

class Main {
	public static void main(String[] args) throws Exception {
		new Main().run();
	}
	
	void run() {
		Scanner sc=new Scanner(System.in);
		int N=sc.nextInt();
		int K=sc.nextInt();
		long[] A=new long[N];
		for(int i=0;i<N;++i) A[i]=sc.nextLong();
		--K;
		if(A[K]==0){
			System.out.println(0);
			return;
		}
		int r=K;
		int l=K;
		while(r+1<A.length&&A[r+1]>=1&&(r==K||A[r]>=2))++r;
		while(l-1>=0&&A[l-1]>=1&&(l==K||A[l]>=2))--l;
		long rc=0,lc=0;
		for(int i=K+1;i<=r;++i)rc+=A[i];
		for(int i=K-1;i>=l;--i)lc+=A[i];
		if(A[K]<=1){
			System.out.println(A[K]+Math.max(rc,lc));
		}else{
			System.out.println(A[K]+rc+lc);
		}
	}
	
	static void tr(Object... objects) {
		System.out.println(Arrays.deepToString(objects));
	}
	
}