結果

問題 No.2549 Paint Eggs
ユーザー AsahiAsahi
提出日時 2023-11-27 02:06:07
言語 Java21
(openjdk 21)
結果
AC  
実行時間 568 ms / 2,000 ms
コード長 13,294 bytes
コンパイル時間 3,954 ms
コンパイル使用メモリ 98,228 KB
実行使用メモリ 70,496 KB
最終ジャッジ日時 2024-09-26 12:05:47
合計ジャッジ時間 23,921 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 162 ms
41,640 KB
testcase_01 AC 165 ms
41,552 KB
testcase_02 AC 154 ms
41,392 KB
testcase_03 AC 163 ms
41,492 KB
testcase_04 AC 163 ms
41,364 KB
testcase_05 AC 141 ms
40,236 KB
testcase_06 AC 140 ms
40,192 KB
testcase_07 AC 145 ms
40,424 KB
testcase_08 AC 139 ms
40,680 KB
testcase_09 AC 144 ms
40,276 KB
testcase_10 AC 476 ms
65,804 KB
testcase_11 AC 388 ms
64,244 KB
testcase_12 AC 361 ms
58,336 KB
testcase_13 AC 447 ms
64,884 KB
testcase_14 AC 345 ms
56,080 KB
testcase_15 AC 212 ms
43,388 KB
testcase_16 AC 426 ms
60,116 KB
testcase_17 AC 336 ms
56,160 KB
testcase_18 AC 313 ms
53,468 KB
testcase_19 AC 402 ms
58,292 KB
testcase_20 AC 538 ms
69,792 KB
testcase_21 AC 521 ms
70,476 KB
testcase_22 AC 540 ms
69,576 KB
testcase_23 AC 521 ms
69,504 KB
testcase_24 AC 553 ms
69,636 KB
testcase_25 AC 467 ms
68,184 KB
testcase_26 AC 456 ms
67,748 KB
testcase_27 AC 426 ms
67,640 KB
testcase_28 AC 516 ms
70,160 KB
testcase_29 AC 455 ms
67,692 KB
testcase_30 AC 539 ms
69,660 KB
testcase_31 AC 482 ms
67,908 KB
testcase_32 AC 546 ms
70,288 KB
testcase_33 AC 483 ms
68,044 KB
testcase_34 AC 461 ms
68,080 KB
testcase_35 AC 519 ms
69,652 KB
testcase_36 AC 568 ms
70,496 KB
testcase_37 AC 522 ms
69,544 KB
testcase_38 AC 438 ms
67,756 KB
testcase_39 AC 465 ms
67,704 KB
testcase_40 AC 510 ms
70,424 KB
testcase_41 AC 459 ms
69,284 KB
testcase_42 AC 503 ms
70,308 KB
testcase_43 AC 508 ms
69,544 KB
testcase_44 AC 494 ms
70,356 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*; 
import java.io.*;
import java.math.*;
import java.util.stream.*;
import java.util.function.*;

class Main implements Runnable {

    public void solve(){
        int n = in.nextInt() , m = in.nextInt() , k = in.nextInt() ;
        int [] c = in.nextInt(n);
        long [] A = in.nextLong(m);
        List<Integer> [] color = ArrayList(m);
        long ans = lnf ;
        for(int i = 0 ; i < n ; i ++) color[c[i] - 1].add(i);
        for(int i = 0 ; i < m ; i ++) {
            ans = Math.min(ans , A[i] * k - A[i] * lower_higher_(color[i] , 0 , k - 1));
            for(int j = 0 ; j < color[i].size() ; j ++) {
                if(color[i].get(j) + (k - 1) < n) ans = Math.min(ans , A[i] * k - A[i] * lower_higher_(color[i] , color[i].get(j) , color[i].get(j) + (k - 1)));
                if(0 <= (color[i].get(j) - (k - 1))) ans = Math.min(ans , A[i] * k - A[i] * lower_higher_(color[i] , color[i].get(j) - (k - 1), color[i].get(j)));
            }
        }
        print(ans);
    }   

    public record Pair<T,S> (T fi , S se) { 
        public String toString() { 
            return "("+fi+","+se+")";
        }
    }

    public record Trio<T,S,U>(T fi , S se , U th) { 
        public String toString() { 
            return "("+fi+","+se+","+th+")";
        }
    }

    public PrintWriter out = new PrintWriter(System.out);
    public In in = new In();
    public static final int mod7 = 1000000007;
    public static final int mod9 = 998244353;
    public static final int  inf = (1  << 30);
    public static final long lnf = (1L << 60);
    public static final String yes = "Yes";
    public static final String no  = "No" ;
    public static final int [] dy4 = {-1,0,1,0};
    public static final int [] dx4 = {0,1,0,-1};
    public static final int [] dy8 = {-1,-1,-1,0,1,1,1,0};
    public static final int [] dx8 = {-1,0,1,1,1,0,-1,-1};

    public boolean isOver(int y , int x , int h , int w) { 
        return y < 0 || x < 0 || y >= h || x >= w ;
    }

    public void swap(int [] array , int l , int r) { 
        int tmp = array[l] ; 
        array[l] = array[r] ; 
        array[r] = tmp ; 
    }

    public void swap(long [] array , int l , int r) { 
        long tmp = array[l] ; 
        array[l] = array[r] ; 
        array[r] = tmp ; 
    }
    
    public String swap(String string , int l , int r) { 
        StringBuilder m = new StringBuilder(string) ;
        m.setCharAt(l, string.charAt(r)); m.setCharAt(r, string.charAt(l)); 
        return m.toString(); 
    }

    public String binarytoString(int a , int len) { 
        String b = Integer.toBinaryString(a); 
        while(b.length() < len) b = "0" + b ; 
        return b ; 
    }

    public String binarytoString(long a , int len) { 
        String b = Long.toBinaryString(a); 
        while(b.length() < len) b = "0" + b ; 
        return b ; 
    }

    public <T extends Comparable<T>> int lower_count(List<T> A , T key) { 
        return upperbound(A, key); 
    }

    public <T extends Comparable<T>> int lowercount(List<T> A , T key) { 
        return lowerbound(A, key); 
    }

    public <T extends Comparable<T>> int higher_count(List<T> A , T key) { 
        return A.size() - lowerbound(A, key); 
    }

    public <T extends Comparable<T>> int highercount(List<T> A , T key) { 
        return A.size() - upperbound(A, key); 
    }

    public <T extends Comparable<T>> int lower_higher_(List<T> A, T x, T y) {
        return upperbound(A, y) - lowerbound(A, x) ;
    }


    private <T extends Comparable<T>> int lowerbound(List<T> A, T key) {
        int left = 0 , right = A.size();
        while (left < right) {
            int mid = (left + right) / 2;
            if (A.get(mid).compareTo(key) < 0) left = mid + 1;
            else right = mid;
        }
        return right;
    }

    private <T extends Comparable<T>> int upperbound(List<T> A, T key) { 
        int left = 0 , right = A.size();
        while (left < right) {
            int mid = (left + right) / 2;
            if (A.get(mid).compareTo(key) <= 0) left = mid + 1;
            else right = mid;
        }
        return right;
    }

    public Integer [] toInteger(int [] a) {
        return Arrays.stream(a).boxed().toArray(Integer[]::new);
    }

    public Long [] toLong(long [] a) {
        return Arrays.stream(a).boxed().toArray(Long[]::new);
    }

    public Double [] toDouble(double [] a) {
        return Arrays.stream(a).boxed().toArray(Double[]::new);
    }

    public <T> List<T> ArrayList(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(ArrayList::new));
    }

    public <T> HashSet<T> HashSet(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(HashSet::new));
    }

    public <T> TreeSet<T> TreeSet(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(TreeSet::new));
    }

    public <T> Deque<T> Deque(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(ArrayDeque::new));
    }

    public <T> PriorityQueue<T> PriorityQueue(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(PriorityQueue::new));
    }

    public <T> List <T> [] ArrayList(int n) { 
        @SuppressWarnings("unchecked")
        List<T> [] G = new ArrayList[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new ArrayList<>();
        return G ; 
    }

    public <T> HashSet <T> [] HashSet(int n) { 
        @SuppressWarnings("unchecked")
        HashSet<T> [] G = new HashSet[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new HashSet<>()  ;
        return G ; 
    }

    public <T> TreeSet <T> [] TreeSet(int n) { 
        @SuppressWarnings("unchecked")
        TreeSet<T> [] G = new TreeSet[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeSet<>()  ; 
        return G ; 
    }

    public <T> TreeSet <T> [] TreeSet(int n , Comparator<? super T> comparator) { 
        @SuppressWarnings("unchecked")
        TreeSet<T> [] G = new TreeSet[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeSet<>(comparator)  ;
        return G ; 
    }

    public <T> Deque <T> [] Deque(int n) { 
        @SuppressWarnings("unchecked")
        Deque<T> [] G = new ArrayDeque[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new ArrayDeque<>(); 
        return G ; 
    }

    public <T> PriorityQueue <T> [] PriorityQueue(int n) { 
        @SuppressWarnings("unchecked")
        PriorityQueue<T> [] G = new PriorityQueue[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new PriorityQueue<>(); 
        return G ; 
    }

    public <T> PriorityQueue <T> [] PriorityQueue(int n , Comparator<? super T> comparator) { 
        @SuppressWarnings("unchecked")
        PriorityQueue<T> [] G = new PriorityQueue[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new PriorityQueue<>(comparator); 
        return G ; 
    }

    public <K,V> HashMap <K,V> [] HashMap(int n) { 
        @SuppressWarnings("unchecked")
        HashMap<K,V> [] G = new HashMap[n];
        for(int i = 0 ; i < n ; i ++ ) G[i] = new HashMap<>()  ;
        return G ; 
    }

    public <K,V> TreeMap <K,V> [] TreeMap(int n) { 
        @SuppressWarnings("unchecked")
        TreeMap<K,V> [] G = new TreeMap[n];
        for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeMap<>()  ;
        return G ;
    }

    public <K,V> TreeMap <K,V> [] TreeMap(int n , Comparator<? super K> comparator) { 
        @SuppressWarnings("unchecked")
        TreeMap<K,V> [] G = new TreeMap[n];
        for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeMap<>(comparator)  ;
        return G ; 
    }

    //---------------------------------------------------------------------------------------//

    @SuppressWarnings("unchecked")
    public <T> void print(T ... data) {
        StringBuilder pr = new StringBuilder();
        for(var temp : data) pr.append(temp+" ");
        out.println(pr.toString());
        out.flush();
    }

    public void print(String [][] array) {
        StringBuilder pr = new StringBuilder();
        for(var temp : array) {
            for(var data : temp) {
                pr.append(data+" ");
            }
            pr.append("\n");
        }
        if(0 < pr.length()) pr.deleteCharAt(pr.length() - 1);
        out.println(pr.toString());
        out.flush();
    }

    public void print(char [][] array) {
        StringBuilder pr = new StringBuilder();
        for(var temp : array) {
            for(var data : temp) {
                pr.append(data);
            }
            pr.append("\n");
        }
        if(0 < pr.length()) pr.deleteCharAt(pr.length() - 1);
        out.println(pr.toString());
        out.flush();
    }

    public <T> void print(Collection<T> collection) {
        StringBuilder pr = new StringBuilder();
        for(T data : collection) pr.append(data +" ");
        out.println(pr.toString());
        out.flush();
    }   

    public static void main(String ... args) {
        new Thread(null, new Main(), "", Runtime.getRuntime().maxMemory()).start(); 
    }

    public void run() {   
        solve();
        out.flush();
    }

}

class In {

    private final InputStream in = System.in;
    private final Scanner sc = new Scanner(System.in);
    private final byte[] buffer = new byte[1024];
    private int ptr = 0;
    private int buflen = 0;

    private boolean hasNextByte() {
        if (ptr < buflen) {
            return true;
        }else{
            ptr = 0;
            try {
                buflen = in.read(buffer);
            } catch (IOException e) {
                e.printStackTrace();
            }
            if (buflen <= 0) {
                return false;
            }
        }
        return true;
    }

    private int readByte() { 
        if (hasNextByte()) return buffer[ptr++]; else return -1;
    }

    private static boolean isPrintableChar(int c) { 
        return 33 <= c && c <= 126;
    }

    private  boolean hasNext() { 
        while(hasNextByte() && !isPrintableChar(buffer[ptr])) {
            ptr++; 
        }
        return hasNextByte();
    }
    
    public String next() {
        if (!hasNext()) throw new NoSuchElementException();
        StringBuilder sb = new StringBuilder();
        int b = readByte();
        while(isPrintableChar(b)) {
            sb.appendCodePoint(b);
            b = readByte();
        }
        return sb.toString();
    }

    public long nextLong() {
        if (!hasNext()) throw new NoSuchElementException();
        long n = 0;
        boolean minus = false;
        int b = readByte();
        if (b == '-') {
            minus = true;
            b = readByte();
        }
        if (b < '0' || '9' < b) {
            throw new NumberFormatException();
        }
        while(true){
            if ('0' <= b && b <= '9') {
                n *= 10;
                n += b - '0';
            }else if(b == -1 || !isPrintableChar(b)){
                return minus ? -n : n;
            }else{
                throw new NumberFormatException();
            }
            b = readByte();
        }
    }

    public int nextInt() {
        long nl = nextLong();
        if (nl < Integer.MIN_VALUE || nl > Integer.MAX_VALUE) throw new NumberFormatException();
        return (int) nl;
    }

    public double nextDouble() { 
        return Double.parseDouble(next());
    }

    public char nextChar() {
        return next().charAt(0);
    }

    public BigInteger nexBigInteger() {
        return sc.nextBigInteger();
    }

    public int [] nextInt(int n) {
        int [] array = new int[n];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextInt();
        }
        return array ;
    }

    public int [][] nextInt(int n , int m) {
        int [][] array = new int[n][m];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextInt(m);
        }
        return array ;
    }

    public long [] nextLong(int n) {
        long [] array = new long[n];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextLong();
        }
        return array ;
    }

    public long [][] nextLong(int n , int m) {
        long [][] array = new long[n][m];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextLong(m);
        }
        return array ;
    }

    public double [] nextDouble(int n) {
        double [] array = new double[n];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextDouble();
        }
        return array ;
    }
    
    public String [] next(int n) {
        String [] array = new String[n];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = next();
        }
        return array ;
    }

    public String [][] next(int n , int m) {
        String [][] array = new String[n][m];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = next(m);
        }
        return array ;
    }

    public char [] nextChar(int n) {
        char [] array = new char[n];
        String string = next() ;
        for(int i = 0 ; i < n ; i ++) {
            array[i] = string.charAt(i);
        }
        return array ;
    }

    public char [][] nextChar(int n , int m) {
        char [][] array = new char[n][m];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextChar(m);
        }
        return array ;
    }

}
0