座位調整
題目描述:
百度辦公區里到處擺放著各種各樣的零食。百度人力資源部的調研發現,員工如果可以在自己喜歡的美食旁邊工作,工作效率會大大提高。因此,百度決定進行一次員工座位的大調整。
調整的方法如下:
1 . 首先將辦公區按照各種零食的擺放分成 N 個不同的區域。(例如:可樂區,餅干區,牛奶區等等)。
2 . 每個員工對不同的零食區域有不同的喜好程度(喜好程度度的范圍為 1 — 100 的整數, 喜好程度越大表示該員工越希望被調整到相應的零食區域)。
3 . 由于每個零食區域可以容納的員工數量有限,人力資源部希望找到一個最優的調整方案令到總的喜好程度最大。
數據輸入:
第一行包含兩個整數 N , M ,( 1<=N , M<=300 )。分別表示 N 個區域和 M 個員工。
第二行是 N 個整數構成的數列 a ,其中 a[i] 表示第 i 個區域可以容納的員工數, (1<=a[i]<=M , a[1]+a[2]+..+a[N]=M) 。
緊接著是一個 M*N 的矩陣 P , P ( i , j )表示第 i 個員工對第 j 個區域的喜好度。
答案輸出:
對于每個測試數據,輸出可以達到的最大的喜好程度。
輸入樣例
3 3
1 1 1
100 50 25
100 50 25
100 50 25
輸出樣例
175
static const int M = 3;
static const int N = 3;
static int capac[] = {1, 1, 1};
vector<int> eraseRow(const vector<int> &v, int i, int currentArea)
{
int colSize = N - currentArea;
vector<int> newVec;
newVec.reserve(v.size() - colSize);
newVec.insert(newVec.begin(),v.begin(),v.begin()+i*colSize);
newVec.insert(newVec.end(),v.begin()+(i+1)*colSize,v.end());
return newVec;
}
vector<int> eraseCol(const vector<int> &v, int currentArea)
{
vector<int> newVec;
for (int i = 0; i != v.size(); ++i)
{
if (i%(N-currentArea))
{
newVec.push_back(v[i]);
}
}
return newVec;
}
int getMax(vector<int> v, int numOfPeop, int numOfCapac, int currentArea = 0, int start = 0)
{
int sum = 0, max = 0;
int curr = currentArea;
if (numOfPeop < numOfCapac)
{
cerr<<"the numOfPeople is smaller than Capacity"<<endl;
return -1;
}
else if (numOfPeop == numOfCapac)
{
for (int i = 0; i != numOfCapac; ++i)
{
max += v[i];
}
}
else
{
vector<int> newVec;
if (numOfCapac == 1)
{
for (int i = 0; i <= numOfPeop - numOfCapac; ++i)
{
sum = v[i * (N - curr)];
newVec = eraseRow(v,i,curr);
newVec = eraseCol(newVec,curr);
if (curr < N)
{
sum += getMax(newVec, numOfPeop -1, capac[curr+1],curr+1);
}
max = (max > sum ? max:sum);
}
}
else
{
if (numOfCapac<1)
{
cerr<<"numOfCapacity is smaller than zero"<<endl;
return -1;
}
for (int i = start; i != numOfPeop - numOfCapac; ++i)
{
sum = v[i * (N - curr)];
newVec = eraseRow(v,i,curr);
sum += getMax(newVec, numOfPeop -1, numOfCapac - 1, curr, i);
max = (max > sum ? max:sum);
}
}
}
return max;
}
void printVec(const vector<int>& v)
{
int delim = 1;
for (int i = 0; i != v.size(); ++i, ++delim)
{
cout<<v[i]<<"\t";
if (!(delim %3))
{
cout<<endl;
}
}
}
void main(int argc, char **argv)
{
ifstream ifs("IntrestArea.txt");
vector<int> v;
v.reserve(M*N);
int i;
while(ifs>>i)
{
v.push_back(i);
}
int max = getMax(v,M,capac[0]);
cout<<max<<endl;
}
多積累函數,才是最寶貴的。。。。并不是 要自己一定能夠編寫出來喲!!!多讀讀