MyCodeSchool Problem #38 Solution (http://www.mycodeschool.com/Problems/Show?problemId=38)

roworcol.cc

#include <iostream>
#include <string>

using namespace std;

class RowOrColSum {
  public:
  long long sumRowOrColumn(long, long, long, char);
};

long long RowOrColSum::sumRowOrColumn(long n, long w, long p, char dir) {
  long long sum = 0;
  if(dir == 'R') {
    long start = p*w - w + 1;
    for(long i = start; i < start + w; ++i) {
      if(i <= n) {
        sum += i;
      } else {
        break;
      }
    } 
  } else if(dir == 'C') {
    for(long i = p; true; i += w) {
      if(i <= n) {
        sum += i;
      } else {
        break;
      }
    }
  }  
  return sum;
}

int main() {
  RowOrColSum obj;

  int cases;
  cin >> cases;

  for(int i = 0; i < cases; ++i) {
    long n, w;
    cin >> n >> w;
    
    long p;
    char dir;
    cin >> p >> dir;

    cout << obj.sumRowOrColumn(n, w, p, dir) << endl;
  }

  return 0;
}