Two Sum
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int nums_size = nums.size();
if(nums_size < 2)
{
printf("The size of nums is less than 2 !!!\n");
}
vector<int> res;
res.push_back(0);
res.push_back(0);
for(int i=0; i<nums_size; i++)
{
for(int j=i+1; j<nums_size; j++)
{
if(target == nums[j] + nums[i])
{
res[0] = i;
res[1] = j;
return res;
}
}
}
printf("No two sum solution !!!\n");
return res;
}
};
void trimLeftTrailingSpaces(string &input) {
input.erase(input.begin(), find_if(input.begin(), input.end(), [](int ch) {
return !isspace(ch);
}));
}
void trimRightTrailingSpaces(string &input) {
input.erase(find_if(input.rbegin(), input.rend(), [](int ch) {
return !isspace(ch);
}).base(), input.end());
}
vector<int> stringToIntegerVector(string input) {
vector<int> output;
trimLeftTrailingSpaces(input);
trimRightTrailingSpaces(input);
input = input.substr(1, input.length() - 2);
stringstream ss;
ss.str(input);
string item;
char delim = ',';
while (getline(ss, item, delim)) {
output.push_back(stoi(item));
}
return output;
}
int stringToInteger(string input) {
return stoi(input);
}
string integerVectorToString(vector<int> list, int length = -1) {
if (length == -1) {
length = list.size();
}
if (length == 0) {
return "[]";
}
string result;
for(int index = 0; index < length; index++) {
int number = list[index];
result += to_string(number) + ", ";
}
return "[" + result.substr(0, result.length() - 2) + "]";
}
int main() {
string line;
while (getline(cin, line)) {
vector<int> nums = stringToIntegerVector(line);
getline(cin, line);
int target = stringToInteger(line);
vector<int> ret = Solution().twoSum(nums, target);
string out = integerVectorToString(ret);
cout << out << endl;
}
return 0;
}
输入:[0,5,4,3,2,4] 6
输出:[2, 4]