Cod sursa(job #2484061)

Utilizator MihclerioVladimir Chim Mihclerio Data 30 octombrie 2019 17:31:29
Problema Party Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.09 kb
#include<bits/stdc++.h>

#define all(s) s.begin(),s.end()
#define rc(x) return cout<<x<<endl,0
#define forn(i,n) for(int i=0;i<int(n);i++)

#define pb push_back
#define mp make_pair
#define fr first
#define sc second

typedef long long ll;
typedef long double ld;

const int nmax=103;
const int mod=998244353;
const ll inf=0x3f3f3f3f3f3f3f3f;

using namespace std;

int n,m,aux;
vector<int> v1(103,0),v2(103,0),v3(103,0);
bool ok,f[103];

bool verif()
{
  for(int i=1;i<=m;i++)
  {
    if(v3[i]==0)
    {
      if(!f[v1[i]] && !f[v2[i]]) return 0;
    } else
    if(v3[i]==1)
    {
      if(!f[v1[i]] && f[v2[i]]) return 0;
    } else
    if(v3[i]==2)
    {
      if(f[v1[i]] && !f[v2[i]]) return 0;
    } else
    if(v3[i]==3)
    {
      if(f[v1[i]] && f[v2[i]]) return 0;
    }
  }
  return 1;
}

int schimba()
{
  for(int i=1;i<=m;i++)
  {
    if(v3[i]==0)
    {
      if(!f[v1[i]] && !f[v2[i]])
      {
        int k=rand()%2;
        f[v1[i]]=k;
        f[v2[i]]=!k;
        return 0;
      }
    } else
    if(v3[i]==1)
    {
      if(!f[v1[i]] && f[v2[i]])
      {
        int k=rand()%2;
        f[v1[i]]=k;
        f[v2[i]]=!k;
        return 0;
      }
    } else
    if(v3[i]==2)
    {
      if(f[v1[i]] && !f[v2[i]])
      {
        int k=rand()%2;
        f[v1[i]]=k;
        f[v2[i]]=!k;
        return 0;
      }
    } else
    if(v3[i]==3)
    {
      if(f[v1[i]] && f[v2[i]])
      {
        int k=rand()%2;
        f[v1[i]]=k;
        f[v2[i]]=!k;
        return 0;
      }
    }
  }
}

void back(int k)
{
  if(!ok)
  {
    if(verif())
    {
      ok=true;
      aux=0;
      for(int i=1;i<=n;i++) aux+=f[i];
      cout<<aux<<"\n";
      for(int i=1;i<=n;i++) if(f[i]) cout<<i<<"\n";
    } else
    {
      schimba();
      back(1);
    }
  }
}

int main()
{
  freopen("party.in","r",stdin);
  freopen("party.out","w",stdout);
  ios_base::sync_with_stdio(0); cin.tie(0);
  cin>>n>>m;
  for(int i=1; i<=m; i++) cin>>v1[i]>>v2[i]>>v3[i];
  for(int i=1;i<=n;i++) f[i]=1;
  ok=false; aux=n;
  back(1);
}