CSES - Datatähti 2017 alku - Results
Submission details
Task:Maalarit
Sender:kh
Submission time:2016-10-16 21:29:22 +0300
Language:Java
Status:READY
Result:68
Feedback
groupverdictscore
#1ACCEPTED12
#2ACCEPTED25
#3ACCEPTED31
#40
Test results
testverdicttimegroup
#1ACCEPTED0.25 s1details
#2ACCEPTED0.15 s1details
#3ACCEPTED0.13 s1details
#4ACCEPTED0.23 s1details
#5ACCEPTED0.26 s1details
#6ACCEPTED0.14 s1details
#7ACCEPTED0.28 s2details
#8ACCEPTED0.28 s2details
#9ACCEPTED0.25 s2details
#10ACCEPTED0.15 s2details
#11ACCEPTED0.26 s2details
#12ACCEPTED0.14 s2details
#13ACCEPTED0.26 s3details
#14ACCEPTED0.28 s3details
#15ACCEPTED0.25 s3details
#16ACCEPTED0.26 s3details
#17ACCEPTED0.26 s3details
#18ACCEPTED0.14 s3details
#19--4details
#20--4details
#21--4details
#22--4details
#23--4details
#24ACCEPTED0.92 s4details

Code

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Scanner;
public class Maalarit2 {
public static void main(String[] args) {
Scanner lukija = new Scanner(System.in);
HashMap<Integer, Lauta> paikat = new HashMap<>();
ArrayList<Lauta> M1 = new ArrayList<>();
ArrayList<Lauta> M2 = new ArrayList<>();
ArrayList<Lauta> laudat = new ArrayList<>();
int maara = Integer.parseInt(lukija.nextLine());
for (int i = 1; i <= maara; i++) {
int korkeus = lukija.nextInt();
Lauta l = new Lauta(i, korkeus);
laudat.add(l);
paikat.put(i, l);
}
Collections.sort(laudat);
int kalleinPaikka = laudat.get(0).getPaikka();
int jakojaannos1 = kalleinPaikka % 2;
for (int i = 1; i <= paikat.size(); i++) {
Lauta l = paikat.get(i);
int Jakojaannos2 = i % 2;
if (Jakojaannos2 == jakojaannos1) {
M1.add(l);
} else {
M2.add(l);
}
}
Collections.sort(M1);
Collections.sort(M2);
int Max1 = M1.get(0).getKorkeus();
int Max2 = M2.get(0).getKorkeus();
int maxPalkkio = Max1 + Max2;
boolean vierekkain = false;
for (Lauta l : M1) {
int korkeus1 = l.getKorkeus();
int paikka1 = l.getPaikka();
if (korkeus1 < Max2) {
break;
}
if (paikka1 == maara) {
if (paikat.get(paikka1 - 1).getKorkeus() >= Max2) {
vierekkain = true;
break;
}
} else if (paikka1 == 1) {
if (paikat.get(paikka1 + 1).getKorkeus() >= Max2) {
vierekkain = true;
break;
}
} else if (paikat.get(paikka1 + 1).getKorkeus() >= Max2 || paikat.get(paikka1 - 1).getKorkeus() >= Max2) {
vierekkain = true;
break;
}
}
if (vierekkain) {
System.out.println(maxPalkkio + " " + 2);
for (int i = 1; i <= maara; i++) {
if (i % 2 == 1) {
System.out.print(1 + " ");
} else {
System.out.print(2 + " ");
}
}
System.out.println("");
} else {
Maalari m1 = new Maalari(1);
Maalari m2 = new Maalari(2);
Maalari m3 = new Maalari(3);
laudat.stream().forEach((l) -> {
if (l.getPaikka() % 2 == jakojaannos1) {
l.setMaalari(m1);
m1.lisaaLauta(l);
} else {
l.setMaalari(m2);
m2.lisaaLauta(l);
}
});
boolean maalareita3 = false;
int alaraja;
int ylaraja;
ArrayList<Integer> kiinnitetytPaikat = new ArrayList<>();
int paikka1;
kiinnitetytPaikat.add(laudat.get(0).getPaikka());
ArrayList<String> paras = new ArrayList<>();
paikat.values().stream().forEach((l) -> {
paras.add(l.getMaalari().getNumero());
});
while (true) {
Lauta max2lauta = m2.getLaudat().get(0);
int max2paikka = max2lauta.getPaikka();
int max2korkeus = max2lauta.getKorkeus();
paikka1 = lahinKiinnitettyEi3Valissa(max2paikka, kiinnitetytPaikat, paikat);
if (paikka1 == -1) {
break;
}
int paikkaero = max2paikka - paikka1;
if (Math.abs(paikkaero) > 1) {
boolean testataan = false;
if (max2paikka == 1) {
if (paikat.get(max2paikka + 1).getKorkeus() < max2korkeus) {
testataan = true;
}
} else if (max2paikka == maara) {
if (paikat.get(max2paikka - 1).getKorkeus() < max2korkeus) {
testataan = true;
}
} else if (paikat.get(max2paikka - 1).getKorkeus() < max2korkeus && paikat.get(max2paikka + 1).getKorkeus() < max2korkeus) {
testataan = true;
}
if (testataan) {
kiinnitetytPaikat.add(max2paikka);
max2lauta.setMaalari(m1);
m1.lisaaLauta(max2lauta);
m2.poistaLauta(max2lauta);
if (paikkaero < 0) {
alaraja = max2paikka;
ylaraja = paikka1;
int kolmosenPaikka = pieninValilla(alaraja, ylaraja, paikat);
Lauta la = paikat.get(kolmosenPaikka);
la.setMaalari(m3);
m3.lisaaLauta(la);
m1.poistaLauta(la);
m2.poistaLauta(la);
int jakojaannos = paikka1 % 2;
for (int i = paikka1 - 1; i > kolmosenPaikka; i--) {
Lauta l = paikat.get(i);
if (l.getPaikka() % 2 == jakojaannos) {
l.setMaalari(m1);
m1.lisaaLauta(l);
m2.poistaLauta(l);
} else {
l.setMaalari(m2);
m2.lisaaLauta(l);
m1.poistaLauta(l);
}
}
jakojaannos = max2paikka % 2;
for (int i = kolmosenPaikka - 1; i > max2paikka; i--) {
Lauta l = paikat.get(i);
if (l.getPaikka() % 2 == jakojaannos) {
l.setMaalari(m1);
m1.lisaaLauta(l);
m2.poistaLauta(l);
} else {
l.setMaalari(m2);
m2.lisaaLauta(l);
m1.poistaLauta(l);
}
}
for (int i = max2paikka - 1; i > 0; i--) {
Lauta l = paikat.get(i);
if (l.getMaalari().getNumero().equals("3")) {
break;
}
if (l.getPaikka() % 2 == jakojaannos) {
l.setMaalari(m1);
m1.lisaaLauta(l);
m2.poistaLauta(l);
} else {
l.setMaalari(m2);
m2.lisaaLauta(l);
m1.poistaLauta(l);
}
}
} else {
alaraja = paikka1;
ylaraja = max2paikka;
int kolmosenPaikka = pieninValilla(alaraja, ylaraja, paikat);
Lauta la = paikat.get(kolmosenPaikka);
la.setMaalari(m3);
m3.lisaaLauta(la);
m2.poistaLauta(la);
m1.poistaLauta(la);
int jakojaannos = alaraja % 2;
for (int i = alaraja + 1; i < kolmosenPaikka; i++) {
Lauta l = paikat.get(i);
if (l.getPaikka() % 2 == jakojaannos) {
l.setMaalari(m1);
m1.lisaaLauta(l);
m2.poistaLauta(l);
} else {
l.setMaalari(m2);
m2.lisaaLauta(l);
m1.poistaLauta(l);
}
}
jakojaannos = ylaraja % 2;
for (int i = kolmosenPaikka + 1; i < ylaraja; i++) {
Lauta l = paikat.get(i);
if (l.getPaikka() % 2 == jakojaannos) {
l.setMaalari(m1);
m1.lisaaLauta(l);
m2.poistaLauta(l);
} else {
l.setMaalari(m2);
m2.lisaaLauta(l);
m1.poistaLauta(l);
}
}
for (int i = ylaraja + 1; i <= maara; i++) {
Lauta l = paikat.get(i);
if (l.getMaalari().getNumero().equals("3")) {
break;
}
if (l.getPaikka() % 2 == jakojaannos) {
l.setMaalari(m1);
m1.lisaaLauta(l);
m2.poistaLauta(l);
} else {
l.setMaalari(m2);
m2.lisaaLauta(l);
m1.poistaLauta(l);
}
}
}
long uusiPalkkio = (long) m1.getPalkkio() + m2.getPalkkio() + m3.getPalkkio();
if (uusiPalkkio < maxPalkkio) {
maalareita3 = true;
maxPalkkio = (int) uusiPalkkio;
paras.clear();
paikat.values().stream().forEach((l) -> {
paras.add(l.getMaalari().getNumero());
});
}
} else {
break;
}
} else {
break;
}
}
if (maalareita3) {
System.out.println(maxPalkkio + " " + "3");
} else {
System.out.println(maxPalkkio + " " + "2");
}
paras.stream().forEach((a) -> {
System.out.print(a + " ");
});
System.out.println("");
}
}
private static int pieninValilla(int alaraja, int ylaraja, HashMap<Integer, Lauta> paikat) {
int pienin = 1000000000;
int paikka = 0;
for (int i = alaraja; i <= ylaraja; i++) {
int korkeus = paikat.get(i).getKorkeus();
if (korkeus <= pienin) {
pienin = korkeus;
paikka = i;
}
if (pienin == 1) {
break;
}
}
return paikka;
}
private static int lahinKiinnitettyEi3Valissa(int max2paikka, ArrayList<Integer> kiinnitetytPaikat, HashMap<Integer, Lauta> paikat) {
ArrayList<Integer> ehdokkaat = new ArrayList<>();
Collections.sort(kiinnitetytPaikat);
int pienilahin = 0;
int suurilahin = 0;
for (int a : kiinnitetytPaikat) {
if (a < max2paikka) {
pienilahin = a;
} else {
suurilahin = a;
break;
}
}
if (pienilahin > 0) {
ehdokkaat.add(pienilahin);
}
if (suurilahin > 0) {
ehdokkaat.add(suurilahin);
}
if (ehdokkaat.size() == 1) {
return ehdokkaat.get(0);
} else {
for (int i = pienilahin + 1; i < max2paikka; i++) {
if (paikat.get(i).getMaalari().getNumero().equals("3")) {
for (int j = max2paikka + 1; i < suurilahin; i++) {
if (paikat.get(j).getMaalari().getNumero().equals("3")) {
return -1;
}
}
return suurilahin;
}
}
return pienilahin;
}
}
private static class Lauta implements Comparable<Lauta> {
private final int paikka;
private final int korkeus;
private Maalari maalari;
private Lauta(int paikka, int korkeus) {
this.paikka = paikka;
this.korkeus = korkeus;
}
private void setMaalari(Maalari m) {
this.maalari = m;
}
private Maalari getMaalari() {
return this.maalari;
}
private int getKorkeus() {
return this.korkeus;
}
private int getPaikka() {
return this.paikka;
}
@Override
public int compareTo(Lauta t) {
int ero = t.getKorkeus() - this.korkeus;
if (ero != 0) {
return t.getKorkeus() - this.korkeus;
} else {
return this.paikka - t.getPaikka();
}
}
}
private static class Maalari {
private final ArrayList<Lauta> OmatLaudat;
private int palkkio;
private final int numero;
private Maalari(int numero) {
this.OmatLaudat = new ArrayList<>();
this.numero = numero;
}
private int getPalkkio() {
if (OmatLaudat == null) {
return 0;
}
Collections.sort(OmatLaudat);
this.palkkio = OmatLaudat.get(0).getKorkeus();
return this.palkkio;
}
private void lisaaLauta(Lauta l) {
if (!this.OmatLaudat.contains(l)) {
this.OmatLaudat.add(l);
}
}
private ArrayList<Lauta> getLaudat() {
Collections.sort(OmatLaudat);
return this.OmatLaudat;
}
private String getNumero() {
return Integer.toString(this.numero);
}
private void poistaLauta(Lauta la) {
OmatLaudat.remove(la);
}
}
}

Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
10
22 54 3 91 69 90 40 29 83 71

correct output
174 3
2 1 2 1 2 1 2 1 2 1 

user output
174 2
2 1 2 1 2 1 2 1 2 1 

Test 2

Group: 1

Verdict: ACCEPTED

input
10
49 3 96 38 90 18 92 74 83 1

correct output
170 3
1 2 1 2 1 2 1 2 1 2 

user output
170 2
1 2 1 2 1 2 1 2 1 2 

Test 3

Group: 1

Verdict: ACCEPTED

input
10
46 3 41 30 16 17 12 93 80 81

correct output
173 3
2 1 2 1 2 1 2 1 2 1 

user output
173 2
1 2 1 2 1 2 1 2 1 2 

Test 4

Group: 1

Verdict: ACCEPTED

input
10
46 8 95 85 82 73 82 92 53 90

correct output
187 3
1 2 1 2 1 2 1 2 1 2 

user output
187 2
1 2 1 2 1 2 1 2 1 2 

Test 5

Group: 1

Verdict: ACCEPTED

input
10
41 18 61 59 40 96 5 2 74 38

correct output
159 3
2 1 2 1 2 1 2 3 1 2 

user output
159 3
2 1 2 1 2 1 2 3 1 2 

Test 6

Group: 1

Verdict: ACCEPTED

input
10
1 1 1 1 1 1 1 1 1 1

correct output
2 3
2 1 2 1 2 1 2 1 2 1 

user output
2 2
1 2 1 2 1 2 1 2 1 2 

Test 7

Group: 2

Verdict: ACCEPTED

input
100
1 39 94 5 24 84 84 10 78 61 38...

correct output
193 3
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

user output
193 2
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

Test 8

Group: 2

Verdict: ACCEPTED

input
100
31 73 18 88 49 28 66 5 32 48 9...

correct output
199 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

user output
199 2
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

Test 9

Group: 2

Verdict: ACCEPTED

input
100
45 56 36 60 31 10 23 79 29 17 ...

correct output
198 3
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

user output
198 2
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

Test 10

Group: 2

Verdict: ACCEPTED

input
100
1 77 70 62 21 68 40 54 90 62 1...

correct output
194 3
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

user output
194 2
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

Test 11

Group: 2

Verdict: ACCEPTED

input
100
4 47 41 81 56 64 12 10 20 100 ...

correct output
189 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

user output
189 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

Test 12

Group: 2

Verdict: ACCEPTED

input
10
1 1 1 1 1 1 1 1 1 1

correct output
2 3
2 1 2 1 2 1 2 1 2 1 

user output
2 2
1 2 1 2 1 2 1 2 1 2 

Test 13

Group: 3

Verdict: ACCEPTED

input
100
256160448 813097800 167146270 ...

correct output
1929869257 3
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

user output
1929869257 2
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

Test 14

Group: 3

Verdict: ACCEPTED

input
100
520002672 3542567 24668528 959...

correct output
1946957555 3
1 2 3 1 2 1 2 1 2 1 2 1 2 1 2 ...

user output
1946957555 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

Test 15

Group: 3

Verdict: ACCEPTED

input
100
483158423 780224665 844754665 ...

correct output
1959373560 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

user output
1959373560 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

Test 16

Group: 3

Verdict: ACCEPTED

input
100
969647264 128558017 889036329 ...

correct output
1997942264 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

user output
1997942264 2
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

Test 17

Group: 3

Verdict: ACCEPTED

input
100
745018527 400495893 635468795 ...

correct output
1961391143 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

user output
1961391143 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

Test 18

Group: 3

Verdict: ACCEPTED

input
10
1 1 1 1 1 1 1 1 1 1

correct output
2 3
2 1 2 1 2 1 2 1 2 1 

user output
2 2
1 2 1 2 1 2 1 2 1 2 

Test 19

Group: 4

Verdict:

input
100000
197349274 775463806 263930657 ...

correct output
1999942635 3
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

user output
(empty)

Test 20

Group: 4

Verdict:

input
100000
102296405 34648120 320393597 9...

correct output
1999930943 3
2 1 2 1 2 1 2 1 2 1 2 1 2 1 2 ...

user output
(empty)

Test 21

Group: 4

Verdict:

input
100000
781254921 418252056 502363453 ...

correct output
1999987794 3
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

user output
(empty)

Test 22

Group: 4

Verdict:

input
100000
849784881 230439009 455097426 ...

correct output
1999979439 3
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

user output
(empty)

Test 23

Group: 4

Verdict:

input
100000
851456132 13422224 537539701 4...

correct output
1999948226 3
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...

user output
(empty)

Test 24

Group: 4

Verdict: ACCEPTED

input
100000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
2 3
3 1 3 1 3 1 3 1 3 1 3 1 3 1 3 ...

user output
2 2
1 2 1 2 1 2 1 2 1 2 1 2 1 2 1 ...