google code jam exercise——Picking Up Chicks

  这道题是Round 1B 2010的第二道题,第一次看的时候,没看懂题目意思,这一次总算是把题目看懂了。看懂之后,才发现,原来如此简单。

  先说一下题目意思,在一条非常narrow的路上,有一群chicks,以各自的速度朝Barn前进。如果后面一只撞上前面的一只,就要以前面一只的速度前进。现在有一个arm of crane,机械臂,可以瞬间交换两只chick的位置,如果一只速度比较快的A被前面一只B挡住了,就可以交换一下,那么A到B的前面去了,而不用以B的速度前进。要求在一定时间T内,有K只到达barn。问是否可能,如果可能,需要交换的次数最少是多少。

  在题目意思理解清楚之后,就会发现,如果chick的位置离barn比较近,而且它能到达barn,那么后面的即使追上它,也不用交换,他们可以顺序的同时到达barn。但如果靠近barn的chick不能到达,比较远的速度较快,可以到达,那么就需要交换。可以到达的chick只需要与在它前面不能到达的交换,而不需要与能到达的交换。

  这样算法就很简单了,先按距离排序,从离barn比较近的开始,检查是否能到,如果不能到,需要交换,swapcnt+1,如果能到,那么passcnt+1,且totalswapcnt += swapcnt,对于这只可以到的chick,需要交换的次数是它前面所有不能到的chick数。如果passcnt大于等于要求通过的数,那么可以结束。

  代码如下:

#!/usr/bin/python
#encoding:UTF-8
#Filename:AlienLanguage.py

import sys

def solveN(nkbt,pos,speed):
    N = nkbt[0]
    K = nkbt[1]
    B = nkbt[2]
    T = nkbt[3]
    posv = [(pos[i],speed[i]) for i in xrange(N)]
    posv.sort()
    passcnt = 0
    swapcnt = 0
    totalswapcnt = 0
    ok = 0
    for i in xrange(N-1,-1,-1):
        if posv[i][1]*T+posv[i][0]>=B:
            passcnt += 1
            totalswapcnt += swapcnt
        else:
            swapcnt += 1
        if passcnt>=K:
            ok = 1
            break;
    return (ok,totalswapcnt)

inname = "input.txt"
outname = "output.txt"
if len(sys.argv)>1:
    inname = sys.argv[1]
    outname = inname.rstrip(".in")
    outname = outname + ".out"
fin = open(inname,"r")
fout = open(outname,"w")

testCaseNum = int(fin.readline().rstrip("\n"))

for caseNum in xrange(testCaseNum):
    nkbt = [int(s) for s in fin.readline().rstrip("\n").split()]
    pos = [int(s) for s in fin.readline().rstrip("\n").split()]
    speed = [int(s) for s in fin.readline().rstrip("\n").split()]
    (ok,res) = solveN(nkbt,pos,speed)
    answer = "Case #%d: " %(caseNum+1)
    if ok:
        answer += str(res) + "\n"
    else:
        answer += "IMPOSSIBLE" + "\n"
    fout.write(answer)

fin.close()
fout.close()

测试通过。

 

posted @ 2012-04-24 20:30  Frandy.CH  阅读(258)  评论(0编辑  收藏  举报