package test;
import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import redis.clients.jedis.Jedis;
public class FindBuyer {
public static void appendMethod(String fileName, String content) {
try {
//打开一个写文件器,构造函数中的第二个参数true表示以追加形式写文件
FileWriter writer = new FileWriter(fileName, true);
writer.write(content);
writer.close();
} catch (IOException e) {
e.printStackTrace();
}
}
public static String[] getStrings() {
FileInputStream fis1;
InputStreamReader isr1;
BufferedReader br1 = null;
try {
fis1 = new FileInputStream("/public/home/dsj/Public/sundujing/test_items.txt");
isr1 = new InputStreamReader(fis1, "UTF-8");
br1 = new BufferedReader(isr1);
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (UnsupportedEncodingException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
String[] strings = new String[5462];
int i=0;
String str;
try {
while ((str = br1.readLine()) != null) {
strings[i] = str;
i++;
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return strings;
}
public static String[] getStrings2() {
FileInputStream fis2;
InputStreamReader isr2;
BufferedReader br2 = null;
try {
fis2 = new FileInputStream("/public/home/dsj/Public/sundujing/user_bought_history.txt");
isr2 = new InputStreamReader(fis2, "UTF-8");
br2 = new BufferedReader(isr2);
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (UnsupportedEncodingException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
String[] strings = new String[13611038];
int i=0;
String str;
try {
while ((str = br2.readLine()) != null) {
strings[i] = str;
i++;
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return strings;
}
/**
* * 二分查找算法 * *
*
* @param srcArray
* 有序数组 *
* @param des
* 查找元素 *
* @return des的数组下标,没找到返回-1
*/
public static int binarySearch1(int[] srcArray, int des){
int low = 0;
int high = srcArray.length-1;
while(low <= high) {
int middle = (low + high)/2;
if(des == srcArray[middle]) {
return middle;
}else if(des <srcArray[middle]) {
high = middle - 1;
}else {
low = middle + 1;
}
}
return -1;
}
public static void main(String[] args)
{
String[] strings = getStrings();//strings[i] test_item_id
String[] strings1=getStrings2();
int[] a=new int[9151109];
String[] a1=new String[9151109];
int[] b=new int[4459929];
String[] b1=new String[4459929];
for(int j=0;j<9151109;j++)
{
String[] str2 = strings1[j].split(" ");
a[j]=Integer.parseInt(str2[1]);
a1[j]=str2[0];
}
for(int k=0;k<4459929;k++)
{
String[] str3 = strings1[k+9151109].split(" ");
b[k]=Integer.parseInt(str3[1]);
b1[k]=str3[0];
}
String fileName = "/public/home/dsj/Public/sundujing/buyers.txt";
String content;
for(int i=0;i<5462;i++)
{
boolean flag=false;
int des=Integer.parseInt(strings[i]);
int index=binarySearch1(a, des);
int index1=binarySearch1(b, des);
if(index!=-1)
{
//strings1[index]up+down
for(int t=index;flag==false;t--)
{
if(a[t]==des)
{
appendMethod(fileName, a1[t]);
appendMethod(fileName, ",");
}else
{
flag=true;
}
}
flag=false;
for(int t=index+1;flag==false;t++)
{
if(a[t]==des)
{
appendMethod(fileName, a1[t]);
appendMethod(fileName, ",");
}else
{
flag=true;
}
}
flag=false;
appendMethod(fileName, "\n");
}else if(index1!=-1)
{
//strings1[index1+9151109]up+down
for(int t=index1;flag==false;t--)
{
if(b[t]==des)
{
appendMethod(fileName, b1[t]);
appendMethod(fileName, ",");
}else
{
flag=true;
}
}
flag=false;
for(int t=index1+1;flag==false;t++)
{
if(b[t]==des)
{
appendMethod(fileName, b1[t]);
appendMethod(fileName, ",");
}else
{
flag=true;
}
}
flag=false;
appendMethod(fileName, "\n");
}else if(index==-1&&index1==-1)
{
appendMethod(fileName, "\n");
}
}
//
}
}
每行为每个预测商品的购买者