#include <iostream>
using namespace std;
int hazard(int n, int kieszonkowe[], string cykl)
{
int czas=0;
int wygrana=0, przegrana=0;
for(int i=0; i<cykl.size(); i++) if(cykl[i]=='W') wygrana++;
else przegrana++;
if(wygrana<przegrana) for(int i=0; i<n; i++)
{
if(i==(n-1)) i=0;
for(int k=0; k<cykl.size(); k++)
{
czas++;
if(cykl[k]=='W') kieszonkowe[i]++;
else if(cykl[k]=='P') kieszonkowe[i]--;
if(kieszonkowe[i]==0) return czas;
}
}
else for(int i=0; i<n; i++)
{
if(kieszonkowe[i]>przegrana) return -1;
else if(kieszonkowe[i]<=przegrana)
{
for(int k=0; k<cykl.size(); k++)
{
czas++;
if(cykl[k]=='W') kieszonkowe[i]++;
else if(cykl[k]=='P') kieszonkowe[i]--;
if(kieszonkowe[i]==0) break;
}
if(kieszonkowe[i]>0) return -1;
}
}
}
int main()
{
int n;
cin>>n;
int kieszonkowe[n];
for(int i=0; i<n; i++)
{
cin>>kieszonkowe[i];
}
int d_cyklu;
string cykl;
cin>>d_cyklu>>cykl;
cout<<hazard(n,kieszonkowe,cykl);
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | #include <iostream> using namespace std; int hazard(int n, int kieszonkowe[], string cykl) { int czas=0; int wygrana=0, przegrana=0; for(int i=0; i<cykl.size(); i++) if(cykl[i]=='W') wygrana++; else przegrana++; if(wygrana<przegrana) for(int i=0; i<n; i++) { if(i==(n-1)) i=0; for(int k=0; k<cykl.size(); k++) { czas++; if(cykl[k]=='W') kieszonkowe[i]++; else if(cykl[k]=='P') kieszonkowe[i]--; if(kieszonkowe[i]==0) return czas; } } else for(int i=0; i<n; i++) { if(kieszonkowe[i]>przegrana) return -1; else if(kieszonkowe[i]<=przegrana) { for(int k=0; k<cykl.size(); k++) { czas++; if(cykl[k]=='W') kieszonkowe[i]++; else if(cykl[k]=='P') kieszonkowe[i]--; if(kieszonkowe[i]==0) break; } if(kieszonkowe[i]>0) return -1; } } } int main() { int n; cin>>n; int kieszonkowe[n]; for(int i=0; i<n; i++) { cin>>kieszonkowe[i]; } int d_cyklu; string cykl; cin>>d_cyklu>>cykl; cout<<hazard(n,kieszonkowe,cykl); } |
English