比賽日程安排 hihoCoder1670
阿新 • • 發佈:2019-02-04
題意:
給M場比賽的時間和參賽的兩支隊伍,判斷這個日程安排是否可行。要求是每支隊伍連續兩場比賽之間至少間隔一天,即每支隊伍一天不能有兩場或更多場比賽,也不能連續兩天都有比賽。
思路:
由於給的資料不是按照日期順序給的,所以我們要先收集好資料在按照日期排個序。收集資料需要注意的是可能同一天有好幾場比賽,如果遇到一支隊伍一天當中有多於一場比賽就可以直接判斷輸出 NO 了。然後判斷從前往後判斷是否是連續的兩天,如果是連續的兩天的話就比較有沒有重的隊伍,如果有相同的隊伍就可以輸出 NO 了。
注意:每支隊伍一天如果有兩場或更多比賽也是不可取的,因為題目裡面說的是“每支隊伍連續兩場比賽之間至少間隔一天”,我就在這裡掉了坑。。。
程式碼:
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<stack>
#include<queue>
#include<utility>
#include<vector>
#include<cmath>
#include<set>
#include<map>
#include<iostream>
#include<algorithm>
#include<sstream>
using namespace std;
typedef long long LL;
int T, N, M;
int t;
bool id[1010][50];
map<pair<int ,int> , int> mp;
vector<pair<int,int> > vec;
bool Judge(pair<int, int> a, pair<int, int> b)
{
if(a.first==b.first && a.second+1==b.second) return true; //月份相同,日期相鄰
else if(a.first==1 && b.first==2 && a.second==31 && b.second==1
|| a.first==2 && b.first==3 && a.second==28 && b.second==1
|| a.first==3 && b.first==4 && a.second==31 && b.second==1
|| a.first==4 && b.first==5 && a.second==30 && b.second==1
|| a.first==5 && b.first==6 && a.second==31 && b.second==1
|| a.first==6 && b.first==7 && a.second==30 && b.second==1
|| a.first==7 && b.first==8 && a.second==31 && b.second==1
|| a.first==8 && b.first==9 && a.second==31 && b.second==1
|| a.first==9 && b.first==10 && a.second==30 && b.second==1
|| a.first==10 && b.first==11 && a.second==31 && b.second==1
|| a.first==11 && b.first==12 && a.second==30 && b.second==1
|| a.first==12 && b.first==1 && a.second==31 && b.second==1) return true; //月份相鄰
return false;
}
int main()
{
//freopen("in.txt", "r", stdin);
scanf("%d", &T);
while(T--){
t = 0;
memset(id, false, sizeof(id));
mp.clear();
vec.clear();
bool ans = false;
scanf("%d%d", &N, &M);
int a, b, c, d;
while(M--){
scanf("%d-%d %d %d", &a, &b, &c, &d);
if(mp.find(make_pair(a,b)) == mp.end()){ //這個日期之前有過
mp[make_pair(a,b)] = ++t;
id[t][c] = true;
id[t][d] = true;
vec.push_back(make_pair(a, b));
}else{ //這個日期之前沒有過,則加入
int tmp = mp[make_pair(a,b)];
if(id[tmp][c]){ //一個隊伍一天有兩場比賽也是要 say NO 的
ans = true;
}else{
id[tmp][c] = true;
}
if(id[tmp][d]){ //一個隊伍一天有兩場比賽也是要 say NO 的
ans = true;
}else{
id[tmp][d] = true;
}
}
}
if(ans) {
printf("NO\n");
continue;
}
sort(vec.begin(), vec.end());
int s = vec.size();
for(int i=0; i<s-1; i++){
if(Judge(vec[i], vec[i+1])){
int tmp1 = mp[vec[i]];
int tmp2 = mp[vec[i+1]];
for(int j=0; j<=N; j++){
if(id[tmp1][j] && id[tmp2][j]){
ans = true;
break;
}
}
}
if(ans) break;
}
if(!ans) printf("YES\n");
else printf("NO\n");
}
return 0;
}