Cod sursa(job #2776347)

Utilizator manciu_ionIon Manciu manciu_ion Data 19 septembrie 2021 14:02:11
Problema Potrivirea sirurilor Scor 18
Compilator java Status done
Runda Arhiva educationala Marime 2.31 kb
//package com.company;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StreamTokenizer;

public class Main {

    public void readData() throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new FileReader("strmatch.in"));
        StreamTokenizer streamTokenizer = new StreamTokenizer(bufferedReader);
        streamTokenizer.nextToken();
        String model = " " + streamTokenizer.sval;
        streamTokenizer.nextToken();
        String strToCompare = " " + streamTokenizer.sval;
        final int[] prefix = makePrefix(model);

        int nrMatches = 0;
        StringBuilder stringBuilder = new StringBuilder();

        for (int i = 1, q = 0; i < strToCompare.length(); i++) {
            while (q > 0 && strToCompare.charAt(i) != model.charAt(q + 1)) {
                q = prefix[q];
            }

            if (strToCompare.charAt(i) == model.charAt(q + 1)) {
                ++q;
            }

            if (q == model.length() - 1) {
                stringBuilder.append(i - model.length() + 1).append(" ");
                nrMatches++;
                q = prefix[model.length() - 1];
                if (nrMatches >= 1000) {
                    break;
                }
            }
        }

        stringBuilder.deleteCharAt(stringBuilder.length() - 1);
        PrintWriter writer = new PrintWriter(new BufferedWriter(new FileWriter("strmatch.out")));
        //System.out.println(nrMatches);
        //System.out.println(stringBuilder.toString());
        writer.println(nrMatches);
        writer.print(stringBuilder.toString());
        writer.close();
    }

    public int[] makePrefix(String model) {
        int[] phi = new int[model.length()];
        phi[1] = 0;
        for (int i = 2, q = 0; i < model.length(); i++) {
            while (q > 0 && model.charAt(q + 1) != model.charAt(i)) {
                q = phi[q];
            }

            if (model.charAt(q + 1) == model.charAt(i)) {
                ++q;
            }
            phi[i] = q;
        }

        return phi;
    }

    public static void main(String[] args) throws IOException {

        new Main().readData();
    }
}