Commit fea9223

authored

Create 2305.Fair-Distribution-of-Cookies_v3.cpp

1 parent f9d6452 commit fea9223Copy full SHA for fea9223

File tree

+52

-0

lines changed

+52

-0

lines changed

Lines changed: 52 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,52 @@`
	`1`	`+class Solution {`
	`2`	`+ int plan[8];`
	`3`	`+public:`
	`4`	`+ int distributeCookies(vector<int>& cookies, int k)`
	`5`	`+ {`
	`6`	`+ sort(cookies.rbegin(), cookies.rend());`
	`7`	`+ int n = cookies.size();`
	`8`	`+`
	`9`	`+ int left = 1, right = INT_MAX;`
	`10`	`+ while (left < right)`
	`11`	`+ {`
	`12`	`+ for (int i=0; i<k; i++)`
	`13`	`+ plan[i] = 0;`
	`14`	`+`
	`15`	`+ int mid = left+(right-left)/2;`
	`16`	`+ if (dfs(cookies, mid, k, 0, (1<<n)-1))`
	`17`	`+ right = mid;`
	`18`	`+ else`
	`19`	`+ left = mid+1;`
	`20`	`+ }`
	`21`	`+ return left;`
	`22`	`+ }`
	`23`	`+`
	`24`	`+ bool dfs(vector<int>& cookies, int limit, int k, int curPerson, int state)`
	`25`	`+ {`
	`26`	`+ if (curPerson == k)`
	`27`	`+ {`
	`28`	`+ return state == 0;`
	`29`	`+ }`
	`30`	`+`
	`31`	`+ for (int subset=state; subset>0; subset=(subset-1)&state)`
	`32`	`+ {`
	`33`	`+ int sum = getSum(cookies, subset);`
	`34`	`+ if (sum > limit) continue;`
	`35`	`+ if (dfs(cookies, limit, k, curPerson+1, state-subset))`
	`36`	`+ return true;`
	`37`	`+ };`
	`38`	`+`
	`39`	`+ return false;`
	`40`	`+ }`
	`41`	`+`
	`42`	`+ int getSum(vector<int>& cookies, int state)`
	`43`	`+ {`
	`44`	`+ int ret = 0;`
	`45`	`+ for (int i=0; i<cookies.size(); i++)`
	`46`	`+ {`
	`47`	`+ if ((state>>i)&1)`
	`48`	`+ ret += cookies[i];`
	`49`	`+ }`
	`50`	`+ return ret;`
	`51`	`+ }`
	`52`	`+};`

Comments

(0)