伪AC自动机

以前一直觉得AC自动机很神奇，总觉得是不是就是自动AC题目的算法

后来学到了才知道是字符串匹配的一个算法

但是还是偶尔去忽悠忽悠小学弟

这次写的一个其实很简单，写的也比较糙

实际上就是模拟提交

不会HTML，所以一边做一边查个个是干嘛的

实际这个小项目就是

1.从POJ爬取题目链接

2.从博客园爬取代码

3.模拟提交

4.判断题目是否AC，如果AC下一题，没有就寻找下一个代码

主要也就是用到了Jsoup，还是挺有意思的一个东西

但是也还是很有问题，就是关于代码提取的问题

因为从博客园提取代码的格式不一样，所以提交上去很有可能是会CE的

以后再改进这个地方，尽量使AC率高些

package locy;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Queue;
import java.util.Scanner;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.commons.codec.EncoderException;
import org.apache.commons.codec.binary.Base64;
import org.jsoup.Connection;
import org.jsoup.Connection.Method;
import org.jsoup.Connection.Response;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;
import org.omg.CORBA.portable.InputStream;

public class Main {
    public static void main(String[] args) throws IOException, EncoderException, InterruptedException {
        ArrayList<String>s = new ArrayList<>();
        Document doc = Jsoup.connect("http://poj.org/problemlist").get();    
        Elements links = doc.select("a[href]");
        Pattern pattern = Pattern.compile("[a-z/:.\?0-9=]+");
        Pattern patternnum = Pattern.compile("[0-9]+");
        Matcher matcher;
        for(Element link:links){
            matcher = pattern.matcher(link.attr("abs:href"));
            if(matcher.matches()){
                matcher = patternnum.matcher(link.text());
                String url = link.attr("abs:href");
                if(matcher.matches())
                    s.add(link.attr("abs:href"));
                
            }
        }
        Map<Integer, String>num = new HashMap<>();
        Map<Integer, String>name = new HashMap<>();
        String cookie = getCookie();
        for(String usl:s){
            FindProblem(usl, name, num,cookie);
        }
    }
    
    /**获取搜索的页面
     * 获取搜索代码的页面
     * @param id
     * @return
     */
    
    public static String getSearchDate(Integer id){
        String date = "http://zzk.cnblogs.com/s/blogpost?Keywords=poj+"+id;
        return date;
    }
    
    
    /**
     * 寻找问题
     * @param url 传入的页面
     * @param nameurl    
     * @param num
     * @throws InterruptedException 
     */
    
    public static void FindProblem(String url,Map<Integer, String>nameurl,Map<Integer,String>num,String coocie) throws InterruptedException{
        try {
            Document doc = Jsoup.connect(url).get();
            Elements links = doc.select("a[href]");
            Pattern pattern = Pattern.compile("[a-z/:.\?0-9=]+");
            Matcher matcher;
            Integer id=null;
            for(Element link:links){
                matcher = pattern.matcher(link.attr("abs:href"));
                if(matcher.matches()&&link.attr("abs:href").length()==30){
                    String str = link.attr("abs:href");
                    id = Integer.parseInt(str.substring(str.length()-4));
                    if(getDate(getSearchDate(id), "poj "+id,coocie,id))
                        System.out.println(id+":true");
                    //getDate(url, name);
                    //getText(url, cookie,a);
                    //nameurl.put(id, link.text());
                    num.put(id, str);
                }
            }
            System.out.println(id+":false");
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }
    
    /**
     * 获取代码
     * @param url
     * @param name
     * @throws InterruptedException 
     */
    
    public static boolean getDate(String url,String name,String cookie,Integer id) throws InterruptedException{
        ArrayList<String>ansurl = new ArrayList<>();
        try {
            Document doc = Jsoup.connect(url).get();
            //System.out.println(url);
            Elements links = doc.select("a[href]");
            for(Element link:links){
                String tempurl = link.attr("abs:href");
                if(tempurl.endsWith(".html")){
                System.out.println(tempurl);
                doc = Jsoup.connect(tempurl).get();
                Elements lin = doc.select("div.cnblogs_code");
                //System.out.println("link"+lin.first().text());   //代码
                try {
                    //System.out.println(lin.text()+"sadf");
                    if(lin.text()==null)
                        continue;
                    if(lin.text().indexOf("#")==-1)
                        continue;
                    push(cookie, lin.text().substring(lin.text().indexOf("#")),id);
                    Thread.sleep(2000);
                    if(getStatu(id.toString()))
                        return true;
                } catch (EncoderException e) {
                    System.out.println("123");
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                }    
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return false;
    }
    
    /**
     * 
     * @param url
     * @param cookie
     * @return 
     * @throws IOException
     * @throws EncoderException
     */
    
    /*
    public static void getText(String url,String cookie,String a) throws IOException, EncoderException{
        Document doc = Jsoup.connect(url).cookie("JSESSIONID",cookie).get();
        //a = getDate(url, name);
        push(cookie,a,id);
        Elements links = doc.select("div");
        for(Element link:links){
            System.out.println(link.className()+"   :"+link.text());
        }
    }*/
    
    
    /**
     * 判断是否提交成功
     * @param id
     * @return
     */
    
    public static boolean getStatu(String id){
        Document doc;
        String flag[] = new String[10];
        try {
            doc = Jsoup.connect("http://poj.org/status").get();
            Elements status = doc.select("tr");
            for(Element sta:status){
                String str = sta.text();
                if(str.indexOf(' ')==8){
                    flag = str.split(" ");
                    System.out.println(flag[3]);
                    System.out.println(sta.text());
                    if(flag[1].equals("flaseacmachine")&&flag[3].equals("Accepted")&&flag[2].equals(id))
                        return true;
                    }
                }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return false;
    }
    
    /**
     * 提交代码
     * @param doc
     * @param url
     * @param cookie
     * @param a
     * @throws IOException
     * @throws EncoderException
     */
    public static void push(String cookie,String a,Integer id) throws IOException, EncoderException{
        System.out.println("sdafasdf");
        Pattern pa = Pattern.compile("[0-9]+[ ]");
        Matcher ma = pa.matcher(a);
        a = ma.replaceAll("").trim();
        int main = a.indexOf("int main()");
        int len = a.indexOf("return 0;",main);
        a=a.substring(0, len+9);
        a+="}";
        Connection.Response re = Jsoup.connect("http://poj.org/submit").
                data("problem_id", id.toString())
                .data("language", "4")
                .data("source",getString(a))
                .data("submit", "Submit")
                .data("encoded", "1").cookie("JSESSIONID",cookie).method(Method.POST).execute();
        }
    
    /**
     * 获取Cookie
     * @return
     * @throws IOException
     */
    
    public static String getCookie() throws IOException{
        
        String surl = "http://poj.org/login";
        
        URL url = new URL(surl);    
        
        HttpURLConnection connection = (HttpURLConnection) url.openConnection();
        
        connection.setDoOutput(true);
        
        OutputStreamWriter out = new OutputStreamWriter(connection.getOutputStream(), "UTF-8");
        
        out.write("user_id1=flaseacmachine&password1=13341352708&B1=login");
        out.flush();
        out.close();
        String cookie = connection.getHeaderField("Set-Cookie");
        //System.out.println(cookie);
        return cookie.substring(11, 33+11);
    }
    
    /**
     * 加密代码
     * @param x
     * @return
     */
    
    public static String getString(String x){
        return new String(Base64.encodeBase64(x.getBytes()));
    }
}