結果

問題 No.2438 Double Least Square
ユーザー chineristACchineristAC
提出日時 2023-08-19 01:00:08
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 2,590 bytes
コンパイル時間 259 ms
コンパイル使用メモリ 82,580 KB
実行使用メモリ 78,596 KB
最終ジャッジ日時 2024-11-28 13:16:16
合計ジャッジ時間 5,607 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
56,372 KB
testcase_01 AC 38 ms
56,372 KB
testcase_02 AC 37 ms
56,164 KB
testcase_03 AC 38 ms
56,572 KB
testcase_04 AC 36 ms
55,680 KB
testcase_05 AC 38 ms
56,272 KB
testcase_06 AC 41 ms
57,196 KB
testcase_07 AC 39 ms
56,436 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 AC 182 ms
78,028 KB
testcase_21 AC 200 ms
77,788 KB
testcase_22 AC 186 ms
77,904 KB
testcase_23 AC 204 ms
78,060 KB
testcase_24 AC 205 ms
77,920 KB
testcase_25 AC 181 ms
77,980 KB
testcase_26 AC 186 ms
77,796 KB
testcase_27 AC 196 ms
77,684 KB
testcase_28 AC 196 ms
77,688 KB
testcase_29 AC 185 ms
78,064 KB
testcase_30 AC 38 ms
57,264 KB
testcase_31 AC 37 ms
56,360 KB
testcase_32 AC 38 ms
56,704 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from itertools import permutations
from heapq import *
from math import acos

input = lambda :sys.stdin.readline().rstrip()
mi = lambda :map(int,input().split())
li = lambda :list(mi())

def solve(N,H,point):

    def calc(f,g):
        res = 0

        a,b,c = 0,0,0
        for i in f:
            x,y = point[i]
            a += x*x
            b += 2*x*(H-y)
            c += (H-y)*(H-y)
        if a!=0:
            t = (-b/(2*a))
            res += a * t * t + b * t + c

        a,b,c = 0,0,0
        for i in g:
            x,y = point[i]
            a += x*x
            b += 2 * x * y
            c += y*y
        if a!=0:
            t = (-b/(2*a))
            res += a * t * t + b * t + c

        return res

    arg = []
    for i in range(N):
        x,y = point[i]
        r = ((y-H/2)**2 + x**2)**.5
        arg.append(acos((H/2-y)/r))
    
    

    idx = [i for i in range(N)]
    idx.sort(key=lambda i:arg[i])

    

    ans = calc([i for i in range(N)],[])
    for i in range(N):
        ans = min(ans,calc([idx[j] for j in range(i+1,N)],[idx[j] for j in range(i+1)]))
        ans = min(ans,calc([idx[j] for j in range(i+1)],[idx[j] for j in range(i+1,N)]))

    return ans

def brute(N,H,point):
    def calc(f,g):
        res = 0

        a,b,c = 0,0,0
        for i in f:
            x,y = point[i]
            a += x*x
            b += 2*x*(H-y)
            c += (H-y)*(H-y)
        if a!=0:
            t = (-b/(2*a))
            res += a * t * t + b * t + c

        a,b,c = 0,0,0
        for i in g:
            x,y = point[i]
            a += x*x
            b += 2 * x * y
            c += y*y
        if a!=0:
            t = (-b/(2*a))
            res += a * t * t + b * t + c

        return res

    ans = 10**100

    for S in range(1<<N):
        f = [i for i in range(N) if S>>i & 1]
        g = [i for i in range(N) if S>>i & 1 == 0]
        ans = min(ans,calc(f,g))


    return ans

import random
while False:
    N = random.randint(3,10)
    H = random.randint(1,10)
    point = []
    for i in range(N):
        x,y = [random.randint(1,10) for _ in range(2)]
        point.append((x,y))
    
    N,H = 3,5
    point = [(7,5),(8,7),(8,2)]
    
    res = solve(N,H,point)
    exp = brute(N,H,point)

    if abs(res-exp) > 10**-2:
        print("WA")
        print(N)
        print(H)
        for x,y in point:
            print(x,y)
        
        print("res:",res)
        print("exp:",exp)
        exit()

    else:
        print("AC")

N = int(input())
H = int(input())
point = [tuple(mi()) for i in range(N)]

print(solve(N,H,point))
0