首页 > 代码库 > Codeforces 727 D T-shirts Distribution

Codeforces 727 D T-shirts Distribution

Description

有 \(6\) 种尺码的衣服,有的人只适合 \(1\) 种衣服,有的人适合相邻的 \(2\) 种,问是否存在合法方案,并输出.

Sol

贪心.

首先,只适合 \(1\) 种衣服的直接减去就好了,如果个数小于 \(0\) 了直接不合法.

然后考虑适合两种衣服的.首先就是,可以跑网络流或者二分图匹配.

一共就有 \(5\) 种情况 \((1,2),(2,3),(3,4),(4,5),(5,6)\) .

很容易的,我们就发现它其实可以贪心...

对于第 \(1\) 种情况,如果 \(1\) 号衣服有剩余的,那么就使用,然后不够就使用下一个即可.

以此类推.

排一下序,统计一下个数直接贪心就好.

Code

#include<cstdio>#include<cstring>#include<utility>#include<algorithm>#include<iostream>using namespace std;#define mpr(a,b) make_pair(a,b)const int N = 100005;int n,m,a[10],b[10];int g[N];pair< pair<int,int> , int>  p[N];pair<int,int> q[10];pair<int,int> ans;int sum[10];int kind(char a[]){	if(a[0]==‘S‘) return 1;	if(a[0]==‘M‘) return 2;	if(a[0]==‘L‘) return 3;	if(strlen(a)==2) return 4;	if(strlen(a)==3) return 5;	return 6;}int main(){	for(int i=1;i<=6;i++) cin>>a[i],b[i]=a[i];	cin>>n;//	cout<<n<<endl;	for(int i=1;i<=n;i++){		char ch=getchar();		char tmp[10];int l=0;		memset(tmp,0,sizeof(tmp));		while(ch>‘Z‘||ch<‘A‘) ch=getchar();		while(ch>=‘A‘&&ch<=‘Z‘) tmp[l++]=ch,ch=getchar();				if(ch==‘,‘){			int t1=kind(tmp),t2;			ch=getchar(),l=0,memset(tmp,0,sizeof(tmp));			while(ch>‘Z‘||ch<‘A‘) ch=getchar();			while(ch>=‘A‘&&ch<=‘Z‘) tmp[l++]=ch,ch=getchar();			t2=kind(tmp);			if(t1>t2) swap(t1,t2);			p[++m]=mpr(mpr(t1,t2),i);		}else{			a[kind(tmp)]--,b[kind(tmp)]--;g[i]=kind(tmp);			if(a[kind(tmp)]<0){ puts("NO");return 0; }		}	}	sort(p+1,p+m+1);		for(int i=1;i<=5;i++) q[i]=mpr(i,i+1);		for(int i=1;i<=m;i++) for(int j=1;j<=5;j++) if(p[i].first==q[j]) sum[j]++;		for(int i=1;i<=6;i++) if(a[i]<0){ puts("NO");return 0; }		for(int i=1;i<=5;i++){		if(a[i]>=sum[i]) a[i]-=sum[i];		else{			a[i+1]-=sum[i]-a[i],a[i]=0;		}		for(int j=1;j<=6;j++) if(a[j]<0){ puts("NO");return 0; }	}	for(int i=1;i<=6;i++) if(a[i]<0){ puts("NO");return 0; }	puts("YES");	for(int i=1;i<=m;i++){		for(int j=1;j<=5;j++) if(p[i].first==q[j]){			if(b[j]>a[j]) g[p[i].second]=j,b[j]--;			else g[p[i].second]=j+1,b[j+1]--;		}	}	for(int i=1;i<=n;i++){		switch(g[i]){			case 1:puts("S");break;			case 2:puts("M");break;			case 3:puts("L");break;			case 4:puts("XL");break;			case 5:puts("XXL");break;			case 6:puts("XXXL");break;		}	}		return 0;}

  

Codeforces 727 D T-shirts Distribution