백준 2866번 문자열 잘라내기 C++

https://www.acmicpc.net/problem/2866

2866번: 문자열 잘라내기

 

루프마다 문자열을 새로 만들면 시간초과가 나서 처음에 전체 문자열을 만들고 substr로 구함

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int r, c;
vector<vector<char>> table;
vector<string> strs;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> r >> c;
    table = vector<vector<char>>(r, vector<char>(c));
    strs = vector<string>(c, "");
    for (int i = 0; i < r; i++) {
        for (int j = 0; j < c; j++) {
            cin >> table[i][j];
        }
    }
    for (int j = 0; j < c; j++) {
        for (int i = 0; i < r; i++) {
            strs[j] += table[i][j];
        }
    }
    bool repeat = true;
    int ans = 0;
    for (int i = 1; i < r; i++) {
        set<string> s;
        for (int j = 0; j < c; j++) {
            string str = strs[j].substr(i, r);
            s.insert(str);
        }
        if (s.size() != c) {
            break;
        }
        ans++;
    }
    cout << ans;
}