I’ve pretty much followed whatever was mentioned in the editorial. My sol’n is passing the sample test case but gives WA upon submission. Please help me rectify the issue, thanks.
#include <iostream>
using namespace std;
int main() {
int cases,d0,d1,d2,d3,d4,d5,d6,digitsum,S;
unsigned long K;
cin >> cases;
while(cases--){
cin >> K >> d0 >> d1;
digitsum = d0+d1;
if(K==2){
(!(digitsum%3))?cout<<"YES"<<endl:cout<<"NO"<<endl;
continue;
}
d2 = (d0+d1)%10;
d3 = (2*(d0+d1))%10;
d4 = (4*(d0+d1))%10;
d5 = (8*(d0+d1))%10;
d6 = (6*(d0+d1))%10;
S = d3+d4+d5+d6;
switch((K-3)%4){
case 0: digitsum += d2+(S*((K-3)/4));
break;
case 1: digitsum += d2+(S*((K-3)/4))+d3;
break;
case 2: digitsum += d2+(S*((K-3)/4))+d3+d4;
break;
case 3: digitsum += d2+(S*((K-3)/4))+d3+d4+d5;
}
(!(digitsum%3))?cout<<"YES"<<endl:cout<<"NO"<<endl;
}
return 0;
}