Java代码实践12306售票算法(二)

/**
* 生成Ticket信息
*
* @param train
* @return
*/
public static List<Ticket> initTicketList(Train train) {
List<Ticket> result = new ArrayList<Ticket>();
Map<String, Integer> seatMap = train.getSeatTotalNum();
for (Entry<String, Integer> entry : seatMap.entrySet()) {
int ticketSize = entry.getValue();
String ticketType = entry.getKey();
for (int i = 0; i < ticketSize; i++) {
int saleChannel = (int) (Math.random() * 10) % 8;
Ticket ticket = new Ticket();
ticket.setSaleChannel(saleChannel);
ticket.setTicketType(ticketType);
ticket.setGuid(UUID.randomUUID().toString());
ticket.setFromDate(train.getFromDate());
ticket.setTicketFlag(CommonUtil.initTicketFlag(train));
ticket.setTrainNo(train.getTrainNo());
result.add(ticket);
}
}
return result;
}

/**
* 创建
* @param i
* @param stationNum
* @return
*/
public static String buidTicket(int i, int stationNum) {
BigInteger temp = new BigInteger("0");
for (int j = i; j < stationNum; j++) {
temp = temp.or(new BigInteger(buidTicket(j)));
}
return temp.shiftRight(1).toString();
}

/**
* 根据筛选条件取得对应的车次
* @param ticketStr
* @param ticketList
* @param condition
* @return
*/
public static Order createOrderByCondition(String ticketStr,List<Ticket> ticketList,Map condition){
Order tempOrder = null;
for (Ticket ticket : ticketList) {
BigInteger toTicket = new BigInteger(ticketStr);
BigInteger fromTicket = new BigInteger(ticket.getTicketFlag());
// 如果可以订票，那么久进行扣除库存&&
// (ticket.getSaleChannel()==(ticket.getSaleChannel()|1))
if (canTicket(fromTicket, toTicket)
&&ticket.getTicketType().equals(condition.get("ticketType").toString())
//&&(ticket.getSaleChannel()==(ticket.getSaleChannel()|2))
) {
tempOrder = new Order();
tempOrder.setOrderId(UUID.randomUUID().toString());
tempOrder.setSeatType(ticket.getTicketType());
tempOrder.setTicketFlag(toTicket.toString());
tempOrder.setTrainNO(ticket.getTrainNo());
tempOrder.setFromDate(ticket.getFromDate());
tempOrder.setSaleChannel(ticket.getSaleChannel());
tempOrder.setTicketGuid(ticket.getGuid());
ticket.setTicketFlag(fromTicket.or(toTicket).toString());
break;
}
}
return tempOrder;
}

/**
* 订票判断是否可以订票
*
* @param fromTicket
* @param toTicket
* @return
*/
private static boolean canTicket(BigInteger fromTicket, BigInteger toTicket) {
return fromTicket.equals(fromTicket.not().or(toTicket).not());
}

package com.train.ticket;
/**
* 订单实体
* @author guo_zhifeng
*
*/
public class Order {

private String orderId;
private String ticketGuid;//票据id
private String ticketFlag;//订票标记
private String seatType;//座位类型
private String fromDate;//发车日期
private String trainNO;//列车编号
private int saleChannel;//销售渠道
public String getOrderId() {
return orderId;
}
public void setOrderId(String orderId) {
this.orderId = orderId;
}
public String getTicketGuid() {
return ticketGuid;
}
public void setTicketGuid(String ticketGuid) {
this.ticketGuid = ticketGuid;
}
public String getTicketFlag() {
return ticketFlag;
}
public void setTicketFlag(String ticketFlag) {
this.ticketFlag = ticketFlag;
}
public String getSeatType() {
return seatType;
}
public void setSeatType(String seatType) {
this.seatType = seatType;
}
public String getFromDate() {
return fromDate;
}
public void setFromDate(String fromDate) {
this.fromDate = fromDate;
}
public String getTrainNO() {
return trainNO;
}
public void setTrainNO(String trainNO) {
this.trainNO = trainNO;
}
public int getSaleChannel() {
return saleChannel;
}
public void setSaleChannel(int saleChannel) {
this.saleChannel = saleChannel;
}
}

package com.train.ticket;
/**
* 票务实体
* @author guo_zhifeng
*
*/
public class Ticket {
private String ticketFlag;
private String ticketType;
private int saleChannel;
private String trainNo;
private String guid;
private String fromDate;//发车日期
public String getGuid() {
return guid;
}
public void setGuid(String guid) {
this.guid = guid;
}
public String getTrainNo() {
return trainNo;
}
public void setTrainNo(String trainNo) {
this.trainNo = trainNo;
}
public String getTicketFlag() {
return ticketFlag;
}
public void setTicketFlag(String ticketFlag) {
this.ticketFlag = ticketFlag;
}
public String getTicketType() {
return ticketType;
}
public void setTicketType(String ticketType) {
this.ticketType = ticketType;
}
public int getSaleChannel() {
return saleChannel;
}
public void setSaleChannel(int saleChannel) {
this.saleChannel = saleChannel;
}
public String getFromDate() {
return fromDate;
}
public void setFromDate(String fromDate) {
this.fromDate = fromDate;
}
}

package com.train.ticket;
import java.util.Map;
/**
* 某一趟的列车的信息
* @author guo_zhifeng
*
*/
public class Train {
private String trainNo;// 火车编号
private int stationNum;// 车站数量
private Map<String, Integer> seatTotalNum;// 各种座位的数量
private String fromDate;
public String getFromDate() {
return fromDate;
}
public void setFromDate(String fromDate) {
this.fromDate = fromDate;
}
public String getTrainNo() {
return trainNo;
}
public void setTrainNo(String trainNo) {
this.trainNo = trainNo;
}
public int getStationNum() {
return stationNum;
}
public void setStationNum(int stationNum) {
this.stationNum = stationNum;
}
public Map<String, Integer> getSeatTotalNum() {
return seatTotalNum;
}
public void setSeatTotalNum(Map<String, Integer> seatTotalNum) {
this.seatTotalNum = seatTotalNum;
}
}

package com.train.main;
import java.io.File;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import com.train.ticket.Order;
import com.train.ticket.Ticket;
import com.train.ticket.Train;
import com.train.util.CommonUtil;
public class MainTest {
public static void main(String[] args) {
Train train = new Train();
train.setTrainNo("SA");
train.setFromDate("//");
train.setStationNum();
Map<String, Integer> seatMap = new HashMap<String, Integer>();
seatMap.put("商务座", );
seatMap.put("一等座", );
seatMap.put("二等座", );
train.setSeatTotalNum(seatMap);
// 生成票据
System.out.println("初始化列车中的票");
List<Ticket> ticketList = CommonUtil.initTicketList(train);
String fileName = "D:\\RESULT.txt";
File f = new File(fileName);
if(f.exists()) f.delete();
long startTime = System.currentTimeMillis();
//int i = ;
//for (Ticket ticket : ticketList) {
// CommonUtil.appendMethodA(fileName,
// i + "||" + CommonUtil.toJSON(ticket) + "\n", true);
// i++;
//}
System.out.println("开始订票");
long beginTime = System.currentTimeMillis();
List<Order> orderResult = new ArrayList<Order>();
for (int j = ; j < train.getStationNum() - ; j++) {
String ticketStr = CommonUtil.buidTicket(j);
//String ticketStr = CommonUtil.buidTicket(,train.getStationNum());;
//System.exit();
List<Order> tempListOrder = CommonUtil.createOrderList(ticketStr,
ticketList, train);
orderResult.addAll(tempListOrder);
}
long endTime = System.currentTimeMillis();
System.out.println("订票完成");
//int m = ;

// for (Ticket ticket : ticketList) {
// String temp = m + "||" + CommonUtil.toJSON(ticket) + ",";
// // System.out.println(temp);
// CommonUtil.appendMethodA(fileName, temp, true);
// m++;
// }
// int k = ;
// for (Order order : orderResult) {
// String temp = order.getOrderId() 
// + "||" + order.getSaleChannel() 
// + "||" + order.getFromDate() 
// + "||" + order.getSeatType() 
// + "||" + order.getTicketGuid()
// + "||" + order.getTrainNO() 
// + "||" + order.getTicketFlag()
// + "||" + new BigInteger(order.getTicketFlag()).toString()
// + "||" +k;
// CommonUtil.appendMethodA(fileName,temp, true);
// k++;
// }

long eedTime = System.currentTimeMillis();
System.out.println("生成订单" + orderResult.size() + "||耗时时间："
+ (endTime - beginTime) + "毫秒");
System.out.println("每秒钟生成单据数(逐张订票)"+ new BigDecimal(orderResult.size()).multiply(new BigDecimal()).divide(new BigDecimal(endTime - beginTime),,BigDecimal.ROUND_HALF_DOWN));
System.out.println("执行完毕");
}
}