結果

問題 No.2408 Lakes and Fish
ユーザー AsahiAsahi
提出日時 2023-08-11 22:02:19
言語 Java21
(openjdk 21)
結果
AC  
実行時間 789 ms / 2,000 ms
コード長 10,207 bytes
コンパイル時間 3,077 ms
コンパイル使用メモリ 92,572 KB
実行使用メモリ 75,572 KB
最終ジャッジ日時 2024-04-29 12:59:27
合計ジャッジ時間 13,997 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 51 ms
37,008 KB
testcase_01 AC 51 ms
37,088 KB
testcase_02 AC 51 ms
36,964 KB
testcase_03 AC 53 ms
36,860 KB
testcase_04 AC 378 ms
54,108 KB
testcase_05 AC 377 ms
57,688 KB
testcase_06 AC 405 ms
55,300 KB
testcase_07 AC 699 ms
74,752 KB
testcase_08 AC 702 ms
74,156 KB
testcase_09 AC 762 ms
75,440 KB
testcase_10 AC 789 ms
75,572 KB
testcase_11 AC 426 ms
56,056 KB
testcase_12 AC 656 ms
68,128 KB
testcase_13 AC 265 ms
48,364 KB
testcase_14 AC 348 ms
54,260 KB
testcase_15 AC 600 ms
67,616 KB
testcase_16 AC 374 ms
55,844 KB
testcase_17 AC 239 ms
45,676 KB
testcase_18 AC 434 ms
60,972 KB
testcase_19 AC 600 ms
62,136 KB
testcase_20 AC 309 ms
49,944 KB
testcase_21 AC 561 ms
63,468 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

class Main{
    // Const //
    static final int MOD7 = 1000000007;
    static final int MOD9 = 998244353 ;
    static final int  inf = (1  << 30);
    static final long lnf = (1L << 60);
    final String yes = "Yes";
    final String no  = "No" ;
    final int [] dy4 = {-1,0,1,0};
    final int [] dx4 = {0,1,0,-1};
    final int [] dy8 = {-1,-1,-1,0,1,1,1,0};
    final int [] dx8 = {-1,0,1,1,1,0,-1,-1};
    // Variables //

    // Solve //
    void solve() {   
        int n = in.nextInt() , m = in.nextInt();
        TreeSet<Long> L = new TreeSet<>();
        TreeSet<Long> R = new TreeSet<>();
        long ans = 0 ;
        for(int i = 0 ; i < n ; i ++ ) {
            L.add(in.nextLong());
        }
        for(long v : L) {
            long l = v - 1 ;
            long r = v + 1 ;
            if(!L.contains(l)) R.add(l);
            if(!L.contains(r)) R.add(r);
        }
        R.add(-1L); R.add((long)1e9 + 1);
        for(int i = 0 ; i < m ; i ++ ) {
            long f = in.nextLong() , b = in.nextLong() , w = in.nextLong();
            if(L.contains(f)) {
                long lc = f - R.lower(f);
                long rc = R.higher(f) - f;
                long cost = Math.min(lc , rc);
                long S_i = b - cost ;
                ans += Math.max(S_i , w);
            }
            else{
                long lc = lnf , rc = lnf ;
                if(L.lower(f) != null) {
                    lc = f - L.lower(f);
                }
                if(L.higher(f) != null) {
                    rc = L.higher(f) - f ;
                }
                long cost = Math.min(lc , rc);
                long S_i = w - cost ;
                ans += Math.max(S_i , b);
            }
        }
        out.println(ans);
    }
    // Function //

    // Main //
    public static void main(String[] args) {
        new Main().solve();
        out.flush();
    }
    // Interface //
    static PrintWriter out = new PrintWriter(System.out);
    static In in = new In();
    static Debug debug = new Debug();
}
// Class //

class Pair<T extends Comparable<T>, U extends Comparable<U>> implements Comparable<Pair<T, U>> {

    private final T first;
    private final U second;

    Pair(T first, U second) {
        this.first = first;
        this.second = second;
    }

    T first() {
        return first;
    }

    U second() {
        return second;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Pair<?, ?> pair = (Pair<?, ?>) o;
        return Objects.equals(first, pair.first) && Objects.equals(second, pair.second);
    }

    @Override
    public int hashCode() {
        return Objects.hash(first, second);
    }

    @Override
    public int compareTo(Pair<T, U> other) {
        T f  = this.first  , F  = other.first;
        U s  = this.second , S  = other.second;
        int First_Result  = f.compareTo(F);
        int Second_Result = s.compareTo(S);
        return First_Result == 0 ? Second_Result : First_Result ;
    }

    @Override
    public String toString() {
        return "(" + this.first + " , " + this.second + ")";
    }
}

class Triple<T extends Comparable<T>, U extends Comparable<U>, V extends Comparable<V>> implements Comparable<Triple<T, U, V>> {
    
    private final T first;
    private final U second;
    private final V third;

    Triple(T first, U second, V third) {
        this.first = first;
        this.second = second;
        this.third = third;
    }

    T first() {
        return first;
    }

    U second() {
        return second;
    }

    V third() {
        return third;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Triple<?, ?, ?> triple = (Triple<?, ?, ?>) o;
        return Objects.equals(first, triple.first) 
            && Objects.equals(second, triple.second)
            && Objects.equals(third, triple.third) ;
    }


    @Override
    public int hashCode() {
        return Objects.hash(first, second, third);
    }

    @Override
    public int compareTo(Triple<T, U, V> other) {
        T f  = this.first  , F  = other.first;
        U s  = this.second , S  = other.second;
        V t  = this.third  , T  = other.third;
        int First_Result  = f.compareTo(F);
        int Second_Result = s.compareTo(S);
        int Third_Result  = t.compareTo(T);
        return Second_Result == 0 ? (First_Result == 0 ? Third_Result : First_Result ) : Second_Result;
    }

    @Override
    public String toString() {
        return "(" + this.first + " , " + this.second + " , " + this.third + ")";
    }
}
// Debug //
class Debug {

    private PrintWriter out = new PrintWriter(System.out);

    public void display(int [] a) { out.println(Arrays.toString(a)); fl();}
    public void display(long [] a) { out.println(Arrays.toString(a)); fl();}
    public void display(double [] a) { out.println(Arrays.toString(a)); fl();}
    public void display(String [] a) { out.println(Arrays.toString(a)); fl();}
    public void display(char [] a) { out.println(Arrays.toString(a)); fl();}
    public void display(boolean [] a) { char [] c = conv(a); display(c); fl();}
    public void display(int [][] a) { for(int i = 0 ; i < a.length ; i ++ ) out.println(Arrays.toString(a[i])); fl();}
    public void display(long [][] a) { for(int i = 0 ; i < a.length ; i ++ ) out.println(Arrays.toString(a[i])); fl();}
    public void display(double [][] a) { for(int i = 0 ; i < a.length ; i ++ ) out.println(Arrays.toString(a[i])); fl();}
    public void display(String [][] a) { for(int i = 0 ; i < a.length ; i ++ ) out.println(Arrays.toString(a[i])); fl();}
    public void display(char [][] a) { for(int i = 0 ; i < a.length ; i ++ ) out.println(Arrays.toString(a[i])); fl();}
    public void display(boolean [][] a) { char [][] c = new char [a.length][a[0].length] ; for(int i = 0 ; i < a.length ; i ++ ) c[i] = conv(a[i]) ; display(c); fl();}
    private char [] conv(boolean [] a) { char [] c = new char[a.length] ; for(int i = 0 ; i < a.length ; i ++ ) c[i] = a[i] ? 'O' : 'X' ; return c ; }
    private void fl() { out.flush(); }
}

// Input // 
class In{

    private final InputStream in = 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;
    }

    public boolean hasNext() { 
        while(hasNextByte() && !isPrintableChar(buffer[ptr])) {
            ptr++; 
        }
        return hasNextByte();
    }

    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();
    }

    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();
        }
    }
    int nextInt() {
        long nl = nextLong();
        if (nl < Integer.MIN_VALUE || nl > Integer.MAX_VALUE) throw new NumberFormatException();
        return (int) nl;
    }
    double nextDouble() { 
        return Double.parseDouble(next());
    }
    char nextChar() {
        return next().charAt(0);
    }
    int [] IntArray(int n) {
        final int [] Array = new int [n];
        for(int i = 0 ; i < n ; i ++ ) {
            Array[i] = nextInt();
        }
        return Array;
    }
    int [][] IntArray(int n , int m) {
        final int [][] Array = new int [n][m];
        for(int i = 0 ; i < n ; i ++ ) {
            Array[i] = IntArray(m);
        }
        return Array;
    }   
    long [] LongArray(int n) {
        final long [] Array = new long [n];
        for(int i = 0 ; i < n ; i ++ ) {
            Array[i] = nextLong();
        }
        return Array;
    }
    long [][] LongArray(int n , int m) {
        final long [][] Array = new long [n][m];
        for(int i = 0 ; i < n ; i ++ ) {
            Array[i] = LongArray(m);
        }
        return Array;
    }
    String [] StringArray(int n) {
        final String [] Array = new String [n];
        for(int i = 0 ; i < n ; i ++ ) {
            Array[i] = next();
        }
        return Array;
    }
    char [] CharArray(int n) {
        final char [] Array = new char[n];
        for(int i = 0 ; i < n ; i ++ ) {
            Array[i] = next().charAt(0);
        }
        return Array;
    }
    char [][] CharArray(int n , int m) {
        final char [][] Array = new char [n][m];
        for(int i = 0 ; i < n ; i ++ ) {
            Array[i] = next().toCharArray();
        }
        return Array;
    }
    char [][] CharArray2(int n , int m) {
        final char [][] Array = new char [n][m];
        for(int i = 0 ; i < n ; i ++ ) {
            for(int j = 0 ; j < n ; j ++ ) {
                Array[i][j] = next().charAt(0);
            }
        }
        return Array;
    }
}
0