Mitrezzz

[АПС] DLLVojska

Nov 27th, 2020
1,046
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. import java.io.BufferedReader;
  2. import java.io.IOException;
  3. import java.io.InputStreamReader;
  4.  
  5. class DLLNode<E> {
  6.     protected E element;
  7.     protected DLLNode<E> pred, succ;
  8.  
  9.     public DLLNode(E elem, DLLNode<E> pred, DLLNode<E> succ) {
  10.         this.element = elem;
  11.         this.pred = pred;
  12.         this.succ = succ;
  13.     }
  14.     @Override
  15.     public String toString() {
  16.         return "<-" + element.toString() + "->";
  17.     }
  18. }
  19.  
  20. class DLL<E> {
  21.     private DLLNode<E> first, last;
  22.  
  23.     public DLL() {
  24.         this.first = null;
  25.         this.last = null;
  26.     }
  27.  
  28.     public void deleteList() {
  29.         first = null;
  30.         last = null;
  31.     }
  32.  
  33.     public int length() {
  34.         int ret;
  35.         if (first != null) {
  36.             DLLNode<E> tmp = first;
  37.             ret = 1;
  38.             while (tmp.succ != null) {
  39.                 tmp = tmp.succ;
  40.                 ret++;
  41.             }
  42.             return ret;
  43.         } else
  44.             return 0;
  45.     }
  46.  
  47.     public void insertFirst(E o) {
  48.         DLLNode<E> ins = new DLLNode<E>(o, null, first);
  49.         if (first == null)
  50.             last = ins;
  51.         else
  52.             first.pred = ins;
  53.         first = ins;
  54.     }
  55.  
  56.     public void insertLast(E o) {
  57.         if (first == null)
  58.             insertFirst(o);
  59.         else {
  60.             DLLNode<E> ins = new DLLNode<E>(o, last, null);
  61.             last.succ = ins;
  62.             last = ins;
  63.         }
  64.     }
  65.  
  66.     public void insertAfter(E o, DLLNode<E> after) {
  67.         if (after == last) {
  68.             insertLast(o);
  69.             return;
  70.         }
  71.         DLLNode<E> ins = new DLLNode<E>(o, after, after.succ);
  72.         after.succ.pred = ins;
  73.         after.succ = ins;
  74.     }
  75.  
  76.     public void insertBefore(E o, DLLNode<E> before) {
  77.         if (before == first) {
  78.             insertFirst(o);
  79.             return;
  80.         }
  81.         DLLNode<E> ins = new DLLNode<E>(o, before.pred, before);
  82.         before.pred.succ = ins;
  83.         before.pred = ins;
  84.     }
  85.  
  86.     public E deleteFirst() {
  87.         if (first != null) {
  88.             DLLNode<E> tmp = first;
  89.             first = first.succ;
  90.             if (first != null) first.pred = null;
  91.             if (first == null)
  92.                 last = null;
  93.             return tmp.element;
  94.         } else
  95.             return null;
  96.     }
  97.  
  98.     public E deleteLast() {
  99.         if (first != null) {
  100.             if (first.succ == null)
  101.                 return deleteFirst();
  102.             else {
  103.                 DLLNode<E> tmp = last;
  104.                 last = last.pred;
  105.                 last.succ = null;
  106.                 return tmp.element;
  107.             }
  108.         }
  109.         return null;
  110.     }
  111.  
  112.     @Override
  113.     public String toString() {
  114.         String ret = new String();
  115.         if (first != null) {
  116.             DLLNode<E> tmp = first;
  117.             ret += tmp + "<->";
  118.             while (tmp.succ != null) {
  119.                 tmp = tmp.succ;
  120.                 ret += tmp + "<->";
  121.             }
  122.         } else
  123.             ret = "Prazna lista!!!";
  124.         return ret;
  125.     }
  126.  
  127.     public DLLNode<E> getFirst() {
  128.         return first;
  129.     }
  130.  
  131.     public void setFirst(DLLNode<E> first) {
  132.         this.first = first;
  133.     }
  134.  
  135.     public void setLast(DLLNode<E> last) {
  136.         this.last = last;
  137.     }
  138.  
  139.     public DLLNode<E> getLast() {
  140.         return last;
  141.     }
  142. }
  143.  
  144. public class DLLVojska {
  145.  
  146.     public static void main(String[] args) throws IOException {
  147.         DLL<Integer> lista = new DLL<Integer>();
  148.         BufferedReader stdin = new BufferedReader(new InputStreamReader(System.in));
  149.         String s = stdin.readLine();
  150.         int N = Integer.parseInt(s);
  151.         s = stdin.readLine();
  152.         String[] ids = s.split(" ");
  153.         for (int i = 0; i < N; i++) {
  154.             lista.insertLast(Integer.parseInt(ids[i]));
  155.         }
  156.  
  157.         s = stdin.readLine();
  158.         String interval[] = s.split(" ");
  159.         int a = Integer.parseInt(interval[0]);
  160.         int b = Integer.parseInt(interval[1]);
  161.  
  162.         s = stdin.readLine();
  163.         interval = s.split(" ");
  164.         int c = Integer.parseInt(interval[0]);
  165.         int d = Integer.parseInt(interval[1]);
  166.  
  167.         DLL<Integer> result = vojska(lista, a, b, c, d);
  168.  
  169.         DLLNode<Integer> node = result.getFirst();
  170.         System.out.print(node.element);
  171.         node = node.succ;
  172.         while (node != null) {
  173.             System.out.print(" " + node.element);
  174.             node = node.succ;
  175.         }
  176.     }
  177.  
  178.     private static DLL<Integer> vojska(DLL<Integer> lista, int a, int b, int c, int d) {
  179.  
  180.         DLLNode<Integer> node = lista.getFirst();
  181.         DLLNode<Integer> start1 = null, start2 = null, end1 = null, end2 = null, tmp;
  182.  
  183.         while (node != null) {
  184.             if (node.element.equals(a)) start1 = node;
  185.             if (node.element.equals(b)) end1 = node;
  186.             if (node.element.equals(c)) start2 = node;
  187.             if (node.element.equals(d)) end2 = node;
  188.             node = node.succ;
  189.         }
  190.  
  191.         // safe because task says we always have a start and an end of both intervals
  192.         start2.pred.succ = start1;
  193.         if (start1 == lista.getFirst()) {
  194.             start1.pred = start2.pred;
  195.             start2.pred = null;
  196.             lista.setFirst(start2);
  197.         } else {
  198.             start1.pred.succ = start2;
  199.             tmp = start1.pred;
  200.             start1.pred = start2.pred;
  201.             start2.pred = tmp;
  202.         }
  203.  
  204.         end1.succ.pred = end2;
  205.         if (end2 == lista.getLast()) {
  206.             end2.succ = end1.succ;
  207.             end1.succ = null;
  208.             lista.setLast(end1);
  209.         } else {
  210.             end2.succ.pred = end1;
  211.             tmp = end1.succ;
  212.             end1.succ = end2.succ;
  213.             end2.succ = tmp;
  214.         }
  215.         return lista;
  216.     }
  217. }
RAW Paste Data