結果

問題 No.742 にゃんにゃんにゃん　猫の挨拶
ユーザー MSKMSK
提出日時 2018-10-10 13:49:39
言語 C++11
(gcc 11.4.0)
結果
TLE  
実行時間 -
コード長 1,507 bytes
コンパイル時間 593 ms
コンパイル使用メモリ 64,428 KB
実行使用メモリ 13,640 KB
最終ジャッジ日時 2024-10-12 17:14:45
合計ジャッジ時間 8,498 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 TLE -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main(void) {
  int N;
  cin >> N;
  vector<int> M(N + 1);
  for(int i = 1; i <= N; i++) {
    cin >> M[i];
  }
  vector<int> V(N + 1, 0);
  for(int i = 1; i <= N; i++) {
    if(0 < M[i] - i) {
      V[i] = 1;
    } else if(0 > M[i] - i) {
      V[i] = -1;
    }
  }
  int maxt = 0;
  for(int i = 1; i <= N; i++) {
    maxt = max(maxt, abs(M[i] - i));
  }
  int ans = 0;
  for(int t = 1; t <= maxt; t++) {
    vector<int> NCatStay(30001, 0);
    vector<int> NCatComeFromL(30001, 0);
    vector<int> NCatComeFromR(30001, 0);
    cerr << t << endl;
    for(int i = 1; i <= N; i++) {
      if(0 == V[i]) {
        cerr << i << ' ' << M[i] << endl;
        NCatStay[M[i]]++;
      } else {
        int pos = i + V[i] * t;
        cerr << i << ' ' << pos << endl;
        if(V[i] > 0) {
          NCatComeFromL[pos]++;
        } else {
          NCatComeFromR[pos]++;
        }
        if(M[i] == pos) {
          V[i] = 0;
        }
      }
    }
    for(int i = 1; i < 30001; i++) {
      int NCatCome = NCatComeFromL[i] + NCatComeFromR[i];
      ans += NCatStay[i] * NCatCome;
      ans += NCatCome * (NCatCome - 1) / 2;
      cerr << i << ' ' << NCatStay[i] << ' ' << NCatCome;
      if(i > 1) {
        ans += NCatComeFromR[i - 1] * NCatComeFromL[i];
        cerr << ' ' << NCatComeFromR[i - 1] << ' '
             << NCatComeFromL[i];
      }
      cerr << ' ' << ans << endl;
    }
  }
  cout << ans << endl;
  return 0;
}
0