Topcoder_3_154

【问题描述】
    有m个工人，有n项任务需要完成，并且已经预知每个工人所能完成的任务。如果工人i完成t项不同的任务，那么必须支付给他cost[i]*t^2的报酬，现在的目的是尽量降低所有任务的总报酬。
    现在给定一个描述每个工人能力的字符串数组，如果第j个位置为'Y'则代表该工人能完成第j项任务，'N'代表不能。同时给定代表每个工人的报酬整数数组cost，第i个值表示第i个工人的基本报酬。要求得到完成所有任务的最低报酬，如果所给条件不能完成所有任务则返回-1。

  定义：
类  JobPlanner
方法  public int minimalCost(String[] can, int[] cost)

  约束：
1、can和cost数组包含相同个数的元素，1至50个；
2、can数组中每个字符串的长度均相同（1至50），并仅由'Y'和'N'组成；
3、cost中的每个元素取值1至500000。

  测试用例：
1、{"YY","YY"}  {1,2}    Returns: 3
2、{"YY","YY"}  {1,5}    Returns: 4
3、{"YN", "YY"}  {1, 5}    Returns: 6
4、{"YN", "YN"}  {1, 1}    Returns: -1

public class JobPlanner { private int minCost = -1; private int totalCount = 0; private boolean[][] canFlag; private int[] jobCost; public int minimalCost(String[] can, int[] cost) { jobCost = cost; canFlag = new boolean[can.length][can[0].length()]; for (int i = 0; i < can.length; i++) { for (int k = 0; k < can[0].length(); k++) { if (can[i].charAt(k) == 'Y') canFlag[i][k] = true; } } realCal(new boolean[canFlag[0].length], new int[canFlag.length]); return minCost; } private void realCal(boolean[] jobFlag, int[] person) { if (totalCount == jobFlag.length) { int sum = 0; for (int i = 0; i < person.length; i++) { sum = sum + jobCost[i] * person[i] * person[i]; } if (minCost == -1 || minCost > sum) minCost = sum; } for (int i = 0; i < jobFlag.length; i++) { if (jobFlag[i] == false) { boolean fix = true; for (int k = 0; k < person.length; k++) { if (canFlag[k][i]) { fix = false; totalCount++; person[k]++; jobFlag[i] = true; realCal(jobFlag, person); jobFlag[i] = false; person[k]--; totalCount--; } } if (fix) return; } } } }