結果

問題 No.5008 [Cherry Alpha] Discrete Pendulum with Air Resistance
コンテスト
ユーザー 👑 Kazun
提出日時 2022-08-06 02:57:11
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 2,850 bytes
コンパイル時間 249 ms
実行使用メモリ 88,784 KB
スコア 0
最終ジャッジ日時 2022-10-14 21:11:23
合計ジャッジ時間 20,529 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other RE * 50
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""
作戦 0: 等差数列で写真を取る
作戦 1: 上位 L 枚を採用
作戦 2: 山登り
作戦 3: アニーリング
作戦 4: 初期解を頑張る.
作戦 5: 作戦 4 の下でアニーリング
作戦 6: ビームサーチ
作戦 7: 全探索
作戦 8: 全探索+アニーリング
"""
#==================================================
# 関数部
def debug(*message):
    print(*message,file=sys.stderr)

def PROBABILITY(delta,temp):
    if delta>=0:
        return 1
    else:
        return exp(delta/temp)

#==================================================
# インポート
from operator import itemgetter
from time import *
from heapq import *
from random import *
from math import *
import sys

#==================================================
# 入力
N,L,Delta=map(int,input().split())

B=[0]*N; M=[0]*N; E=[0]*N
for i in range(N):
    B[i],M[i],E[i]=map(int,input().split())
#==================================================
H=[0]*N; F=[0]*N
for i in range(N):
    H[i]=(B[i]-M[i])//E[i]
    F[i]=2*B[i]*H[i]-E[i]*H[i]*H[i]

#==================================================
BEGIN_TIME=time()
TIME_LIMIT=1.000
SECOND_PHASE=0.800

TEMPERATURE=500
ALPHA=0.999

CHANCE=4

#==================================================
S=[0]
X=[0]*N; A=B.copy(); D=[1]*N
coef=pow(10,7)*2/(N*(N-1))

while time()-BEGIN_TIME<TIME_LIMIT:
    for i in range(N):
        X[i]+=D[i]
        if X[i]==D[i]*A[i]:
            D[i]*=-1
            A[i]=max(M[i],A[i]-E[i])
    s=0
    for i in range(N):
        for j in range(i+1,N):
            s+=abs(X[i]-X[j])/(B[i]+B[j])
    S.append(round(coef*s))

debug("用意できたデータ数:",len(S))

#==================================================
# 初期解の生成

X=[]
for t in range(0,len(S),100):
    X.append((t,S[t]))

K=L
X.sort(key=itemgetter(1),reverse=True)

T=[t for t,_ in  X[:K]]
T=[0]+T+[(len(S)-1)+Delta]
T.sort()
TT=T.copy()

#==================================================

before=0
for t in T[1:-1]:
    before+=S[t]

max_score=-1

for _ in range(CHANCE):
    challenge=0
    BEGIN_TIME=time()
    T=TT.copy()
    while time()-BEGIN_TIME<SECOND_PHASE/CHANCE:
        challenge+=1
        if challenge%100==0:
            TEMPERATURE*=ALPHA

        j=randint(1,K)
        u=T[j]+(1 if random()<0.5 else -1)
        if T[j-1]+Delta<=u<=T[j+1]-Delta and random()<PROBABILITY(S[u]-S[T[j]],TEMPERATURE):
            T[j]=u

    score=0
    for t in T[1:-1]:
        score+=S[t]

    if max_score<score:
        max_score=score
        Ans=T

#==================================================
T=Ans
after=0
for t in T[1:-1]:
    after+=S[t]

#==================================================
debug("変更前:",before)
debug("変更後:",after)
debug("改善率:",after/before)

#==================================================
T=T[1:L+1]
print(K)
print(*T)
0