Tyvj_P1007

描述

上课的时候总有一些同学和前后左右的人交头接耳,这是令小学班主任十分头疼的一件事情。
不过,班主任小雪发现了一些有趣的现象,当同学们的座次确定下来之后,只有有限的D对同学上课时会交头接耳。
同学们在教室中坐成了M行N列,坐在第i行第j列的同学的位置是(i,j),为了方便同学们进出,
在教室中设置了K条横向的通道,L条纵向的通道。于是,聪明的小雪想到了一个办法,
或许可以减少上课时学生交头接耳的问题:她打算重新摆放桌椅,改变同学们桌椅间通道的位置,
因为如果一条通道隔开了两个会交头接耳的同学,那么他们就不会交头接耳了。
请你帮忙给小雪编写一个程序,给出最好的通道划分方案。在该方案下,上课时交头接耳的学生对数最少。
输入格式

输入的第一行,有5各用空格隔开的整数,分别是M,N,K,L,D(2<=N,M<=1000,0<=K

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Scanner;
public class Main {
    private static int M;
    private static int N;
    private static int K;
    private static int L;
    private static int D;
    private static ArrayList<KeyValue> rowArrange=new ArrayList<KeyValue>();
    private static ArrayList<KeyValue> colArrange=new ArrayList<KeyValue>();
    private static Point[] pArr;

    /** * @param args */
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner sc=new Scanner(System.in);
        M=sc.nextInt();
        N=sc.nextInt();
        K=sc.nextInt();
        L=sc.nextInt();
        D=sc.nextInt();
        pArr=new Point[D];
        for(int i=0;i<D;i++){
            pArr[i]=new Point(sc.nextInt(),sc.nextInt(),sc.nextInt(),sc.nextInt());
        }
        for(int i=1;i<M;i++){
            KeyValue kv=new KeyValue(i,0);
            rowArrange.add(kv);
        }
        for(int j=1;j<N;j++){
            KeyValue kv=new KeyValue(j,0);
            colArrange.add(kv);
        }
        for(int i=0;i<D;i++){
            Point p=pArr[i];
            if(p.x==p.p){
                if(p.y<p.q){
                    KeyValue kv=colArrange.get(p.y-1);
                    kv.value++;
                }else{
                    KeyValue kv=colArrange.get(p.q-1);
                    kv.value++;
                }

            }
            if(p.y==p.q){
                if(p.x<p.p){
                    KeyValue kv=rowArrange.get(p.x-1);
                    kv.value++;
                }else{
                    KeyValue kv=rowArrange.get(p.p-1);
                    kv.value++;
                }
            }
        }

        Collections.sort(rowArrange);
        Collections.sort(colArrange);

        int[] rowResult=new int[K];
        int[] colResult=new int[L];
        for(int i=0;i<K;i++){
            rowResult[i]=rowArrange.get(i).key;
        }

        for(int i=0;i<L;i++){
            colResult[i]=colArrange.get(i).key;
        }

        Arrays.sort(rowResult);
        Arrays.sort(colResult);

        for(int i=0;i<K;i++){
            if(i!=K-1){
                System.out.print(rowResult[i]+" ");
            }else{
                System.out.print(rowResult[i]);
            }
        }
        System.out.println();
        for(int i=0;i<L;i++){
            if(i!=L-1){
                System.out.print(colResult[i]+" ");
            }else{
                System.out.print(colResult[i]);
            }
        }

    }
}

class Point{
    int x;
    int y;
    int p;
    int q;
    public Point(int x,int y,int p,int q){
        this.x=x;
        this.y=y;
        this.p=p;
        this.q=q;
    }
}

class KeyValue implements Comparable<KeyValue>{
    int key;
    int value;
    public KeyValue(int k,int v){
        key=k;
        value=v;
    }

    public int compareTo(KeyValue arg0) {
        // TODO Auto-generated method stub
        if(value<arg0.value){
            return 1;
        }else if(value==arg0.value){
            if(key>arg0.key){
                return 1;
            }else{
                return -1;
            }
        }else{
            return -1;
        }
    }   
}

你可能感兴趣的:(Tyvj_P1007)